[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

Pull Request Overview

  • Opened on October 21, 2026
  • Status Merged
  • Commit count 1 with first commit October 21, 2026

Total Delta

0 Total Diff Delta

Open Days

Open 17 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?

Fraction of total time Business days Phase
 
0.0 days Authoring 1 commit before pull request opened for review
 
0.3 days Awaiting first review
 
0.1 days Revising work with 0 commits in response to 1 review that left 1 comment
Set up deploy tracking to begin tracking full PR start-to-deploy times

Total time for pull request 0.4 business days from first commit authored to merge
Deploy tracking has not been set up

Author avatar

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

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.

Comments Threads Pending Resolution

jakevdp reviewed and approved this work on October 21, 2026
jakevdp left a comment

Thanks!

Resolved Comment Threads

No resolved comments have been left on this PR.