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 1 year ago

tengyifei requested a review from

rpsilva-aws 1 year ago

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

tengyifei merged 8e6ca600 into master 1 year ago

tengyifei deleted the yifeit/fix-grad-accumulation branch 1 year ago

Reviewers

rpsilva-aws

Assignees

No one assigned

Labels

None yet

Milestone

No milestone