google/jax
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][NFC] Make compress swaps explicit
[Pallas:SC][NFC] Make compress swaps explicit
SC doesn't support non-compress stores yet, and tpu.vector_store has compressed semantics today.
Although Pallas doesn't expose masked store to non-compress uses today, it's not obvious that _store_lowering_rule supports only compressed semantics.
This CL makes this restriction clear.
Once we add support for non-compress stores, compress will be used to differentiate between tpu.vector_compress_store and tpu.vector_store.
No comments have been left on this PR.