xla
Cherry-pick: Enable xla:gpu autocast for bfloat16 if not restricted (#5570)
#5591
Merged

Cherry-pick: Enable xla:gpu autocast for bfloat16 if not restricted (#5570) #5591

JackCaoG merged 1 commit into r2.1 from wcromar/2.1-cherry-pick-amp-fix
will-cromar
yeounoh Enable xla:gpu autocast for bfloat16 if not restricted (#5570)
4d8d1cc1
will-cromar will-cromar requested a review from yeounoh yeounoh 2 years ago
will-cromar will-cromar requested a review from JackCaoG JackCaoG 2 years ago
JackCaoG
JackCaoG approved these changes on 2023-09-15
JackCaoG JackCaoG merged 1e6c11d1 into r2.1 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone