PR #5337 fix amp dtype setting for GPU

fix amp dtype setting for GPU #5337

JackCaoG merged 4 commits into pytorch:master from baoleai:features/fix_amp

fix amp dtype setting for GPU.

dee2819f

baoleai changed the title ~~fix amp dtype setting for GPU.~~ fix amp dtype setting for GPU 2 years ago

JackCaoG requested a review from

cowanmeg 2 years ago

JackCaoG added xla:gpu

cowanmeg commented on 2023-07-25

fix ut

af805a25

fix lint.

c156f772

JackCaoG requested a review from

cowanmeg 2 years ago

cowanmeg approved these changes on 2023-07-26

minor.

1d40e17d

JackCaoG merged 22ba4af2 into master 2 years ago

Reviewers

cowanmeg

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone