[fuser] Fuser fixes #40774

Merged
copybara-service[bot] opened 3:20pm on September 17, 2026 wanted to merge 91 Δ into google/jax main from
test_982718121

Pull request #40774 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 91
Maximum comment count per PR 10 0
Pull request maximum business days between activity 3 0
Pull request maximum business days open 5 1

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 jax/_src/pallas/fuser/__init__.py
tests/pallas/tpu_fusible_matmul_with_stateful_fusions_test.py
1
100 jax/_src/pallas/fuser/__init__.py
jax/experimental/pallas/fuser.py
1
100 jax/_src/pallas/fuser/__init__.py
jax/_src/pallas/fuser/block_spec.py
1
100 jax/experimental/pallas/fuser.py
jax/_src/pallas/fuser/block_spec.py
1
85 jax/experimental/pallas/fuser.py
tests/pallas/tpu_fusible_matmul_with_stateful_fusions_test.py
1
85 jax/experimental/pallas/fuser.py
jax/_src/pallas/fuser/__init__.py
1
80 jax/_src/pallas/fuser/__init__.py
tests/pallas/tpu_fusible_matmul_test.py
1
71 tests/pallas/tpu_fusible_matmul_with_stateful_fusions_test.py
jax/_src/pallas/fuser/block_spec.py
1

Biggest files changes in pull request

Estimated line counts are shown. Values may change once change summary has finished building.

Path View whole file Lines removed Lines added Final line count
jax/_src/pallas/fuser/block_spec.py 0 50 ---
tests/pallas/tpu_fusible_matmul_test.py 4 3 ---
tests/pallas/tpu_fusible_matmul_with_stateful_fusions_test.py 20 3 ---
jax/_src/pallas/fuser/fusible.py 1 3 ---
jax/_src/pallas/fuser/__init__.py 0 1 ---
jax/experimental/pallas/fuser.py 0 1 ---

️ Biggest functions grown or shrunk by pull request

Which functions had their line count change as part of this pull request?

Stat calculation pending updated commit group build. Please check back in a few minutes.