[pallas:fuser] Do not scalar prefetch inside nested calls in `push_block_spec` #41852

Merged
copybara-service[bot] opened 11:00am on October 21, 2026 wanted to merge 99 Δ into google/jax main from
test_1003941194

Pull request #41852 quality checks

Code files and functions touched by this pull request are analyzed below.

Quality targets

How does this pull request compare to chosen code quality targets? Read more about code quality targets.

Target name Target value This PR
Maximum Diff Delta per PR 500 99
Maximum comment count per PR 10 0
Pull request maximum business days between activity 3 16
Pull request maximum business days open 5 17

Coupled files

Are there any files from this commit that are always committed alongside another file? This often suggests that there exists code coupling to potentially remove. Read more about code coupling

Committed together % Code file
Coupled code file
Commits in PR with both
100 tests/pallas/fuser_block_spec_test.py
jax/_src/pallas/fuser/block_spec.py
1

Biggest files changes in pull request

Path View whole file Lines removed Lines added Final line count
jax/_src/pallas/fuser/block_spec.py 8 16 3761
tests/pallas/fuser_block_spec_test.py 0 33 3280