[pull] master from tensorflow:master#1671

Merged

pull[bot] merged 2 commits into

GesuBackups:masterfrom

tensorflow:master

Mar 29, 2026

Commits on Mar 28, 2026

PR #39382: [xla:gpu] Migrate collective thunks to generic Async Start/Done

ezhulenev
authored and
tensorflower-gardener
committed

Commits on Mar 29, 2026

Rollback to not trigger fatal log on reshard through full rematerialization since there seems to be corner cases that are not covered.

Tongfei-Guo
authored and
tensorflower-gardener
committed