xla
fix amp dtype setting for GPU
#5337
Merged

fix amp dtype setting for GPU #5337

JackCaoG merged 4 commits into pytorch:master from baoleai:features/fix_amp
baoleai
baoleai fix amp dtype setting for GPU.
dee2819f
baoleai baoleai changed the title fix amp dtype setting for GPU. fix amp dtype setting for GPU 2 years ago
JackCaoG JackCaoG requested a review from cowanmeg cowanmeg 2 years ago
JackCaoG JackCaoG added xla:gpu
cowanmeg
cowanmeg commented on 2023-07-25
baoleai fix ut
af805a25
baoleai fix lint.
c156f772
JackCaoG JackCaoG requested a review from cowanmeg cowanmeg 2 years ago
cowanmeg
cowanmeg approved these changes on 2023-07-26
JackCaoG
baoleai minor.
1d40e17d
baoleai
JackCaoG
JackCaoG JackCaoG merged 22ba4af2 into master 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone