DeepSpeed
Relax restrictions of torch.autocast integration
#7543

Merged

Relax restrictions of torch.autocast integration #7543

tohtana merged 3 commits into master from tohtana/loosen_autocast_assertion

tohtana requested a review from

tjruwase 22 days ago

tohtana requested a review from

loadams 22 days ago

loosen restriction of autocast

72dde3c1

tohtana force pushed from b25172cd to 72dde3c1 22 days ago

tohtana changed the title ~~loosen restriction of autocast~~ Relax restrictions of torch.autocast integration 22 days ago

print log only on rank0

502e552a

Merge branch 'master' into tohtana/loosen_autocast_assertion

88c16100

tohtana enabled auto-merge (squash) 22 days ago

disabled auto-merge 22 days ago
Manually disabled by user

tohtana merged 66bf2a64 into master 22 days ago

tohtana deleted the tohtana/loosen_autocast_assertion branch 22 days ago

Reviewers

tjruwase

loadams

Assignees

No one assigned

Labels

None yet

Milestone

No milestone