[Pallas TPU] Add missing f-string prefixes in get/swap lowering error messages #41858

Merged
ayaka14732 opened 12:21pm on October 21, 2026 wanted to merge 1 commit into google/jax main from
pallas-sc-fstring
Diff Delta:
0
Classified as:  Bug Fix

ayaka14732's Description of Work

The SparseCore get/swap lowering rules raise errors with literal {strides} and {out_aval.shape} placeholders because the f prefix is missing, e.g. Get only supports slices with stride 1, got {strides}. This PR adds the prefix to the four affected strings in jax/_src/pallas/mosaic/sc_lowering.py so the actual values are shown.

0 total changed files
Congrats, you've cleared the final file
Like the credits, but with fewer stunts.