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

Closed
copybara-service[bot] opened 11:33am on October 21, 2026 wanted to merge 1 commit into google/jax main from
test_957659209

Pull Request Overview

  • Opened on October 21, 2026
  • Status Closed
  • Commit count 1 with first commit October 23, 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?

Data pending calculation for pull request

Author avatar

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

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.

PR was closed without comments.