Review Roll forward: lower cumulative reductions to chlo.ScanOp on TPU (#41855)

Code line content in this commit was purged after it aged past our code retention window of two weeks for this repo. We have insufficient quota to grab the code content at this time, please try back later.

If you receive this message over several days, please consider adding more users to your account (whose quota we can use to grab commit content), or contact [email protected] with error #33NFNE

copybara-service[bot]

google/jax

test_957659209

Diff Delta:

About 0 Diff Delta/hour

Classified as: General

copybara-service[bot]'s Description of Work

Roll forward: lower cumulative reductions to chlo.ScanOp on TPU

Relands the chlo.ScanOp lowering of cumsum, cumprod, cummax and cummin
on TPU. The downstream issues behind the rollback are fixed. Export mode
and libtpu versions older than 0.0.51 keep the reduce_window lowering;
that guard can be removed after 2026-11-21.

3 total changed files

Loading changes...

You’ve reached the end

Like the credits, but with fewer stunts.

Roll forward: lower cumulative reductions to chlo.ScanOp on TPU #41855

copybara-service[bot]'s Description of Work