Home » Oreps » Jax » All Pull Requests » #40779

google/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

1

Commit Activity

Directory Browser

Velocity History

[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

Pull Request Overview

Opened on September 18, 2026
Status Open
Commit count 1 with first commit September 17, 2026

Total Delta

0 Total Diff Delta

Open Days

Open 0 weekdays

Test Delta

0 Diff Delta in Test Files

Breakdown by Phase

How long has this pull request spent in each phase of its lifecycle?

Data pending calculation for pull request

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

[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.

No comments have been left on this PR.

GitClear uses cookies to ensure you get the best experience on our website. Learn more about our cookie policy