PR #8628 Fix grad accumulation tests on TPUv5/v6

Fix grad accumulation tests on TPUv5/v6 #8628

tengyifei merged 1 commit into master from yifeit/fix-grad-accumulation

Fix grad accumulation tests on TPUv5/v6

92b32de7

tengyifei marked this pull request as ready for review 242 days ago

tengyifei requested a review from

rpsilva-aws 242 days ago

rpsilva-aws approved these changes on 2025-01-27

tengyifei merged 8e6ca600 into master 241 days ago

tengyifei deleted the yifeit/fix-grad-accumulation branch 241 days ago

Reviewers

rpsilva-aws

Assignees

No one assigned

Labels

None yet

Milestone

No milestone