DeepSpeed
Switch from torch.cuda.amp.custom_fwd to torch.amp.custom_fwd(device=...)
#5684

Closed

Switch from torch.cuda.amp.custom_fwd to torch.amp.custom_fwd(device=...) #5684

loadams wants to merge 2 commits into master from loadams/torch-future-warnings

Update autocast to switch from torch.cuda.amp.custom_fwd to torch.amp…

1b6cf603

Merge branch 'master' into loadams/torch-future-warnings

5ce6ef53

loadams closed this 1 year ago

loadams deleted the loadams/torch-future-warnings branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone