Home » Oreps » Jax » All Pull Requests » #40779
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
1
Commit Activity
Pull Requests
Directory Browser
Velocity History
Releases

[Pallas:SC] Lower swaps to tpu.compress_store_vreg or tpu.store without layout passes #40779

Compare code on GitHub
Open
copybara-service[bot] opened 5:05pm on September 17, 2026 wants to merge 1 commit into google/jax main from
test_983470154
copybara-service[bot]
google/jax
test_983470154
Diff Delta:
0
Classified as:  General

copybara-service[bot]'s Description of Work

[Pallas:SC] Lower swaps to tpu.compress_store_vreg or tpu.store without layout passes

On the path that does not run the layout passes, masked (compressed) swaps now lower to tpu.compress_store_vreg instead of a masked tpu.vector_store.

0 total changed files
You are now leaving the Pull Request Diff Zone
The diff well is dry.
GitClear uses cookies to ensure you get the best experience on our website. Learn more about our cookie policy