Remove amp() from abstract accelerator #7879
Replace get_accelerator().amp().autocast() with torch.amp.autocast(de…
2e856639
Remove get_accelerator().amp() API and use torch.amp directly
a5b04f02
Merge branch 'master' into gma/improve_amp
f48e8eb1
Merge branch 'master' into gma/improve_amp
4e7c6f00
update pytorch version in workflow nv-pre-compile-ops.yml
a31a5db5
delock
force pushed
from
12e531bd
to
a31a5db5
15 hours ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub