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

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

Total Delta

99 Total Diff Delta

Open Days

Open 17 weekdays

Test Delta

33 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

Author avatar

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

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

Scalar prefetch values must come from fusion consts, but a nested call receives
all of its inputs as arguments. pull_block_spec already accounts for this,
while push_block_spec failed an assertion on nested ops with dynamic indices,
e.g. dynamic_update_slice, even when the push rule did not need them.

PR was closed without comments.