DeepSpeed
Enable hpz when running with torch.no_grad
#4232
Merged

Enable hpz when running with torch.no_grad #4232

HeyangQin merged 7 commits into master from HeyangQin/enable_hpz_nograd
HeyangQin
HeyangQin enable hpz when running with torch.no_grad
f5defcb9
HeyangQin HeyangQin requested a review from jeffra jeffra 2 years ago
HeyangQin HeyangQin requested a review from tjruwase tjruwase 2 years ago
HeyangQin HeyangQin requested a review from samyam samyam 2 years ago
HeyangQin HeyangQin requested a review from mrwyattii mrwyattii 2 years ago
HeyangQin Merge branch 'master' into HeyangQin/enable_hpz_nograd
3f896625
HeyangQin HeyangQin enabled auto-merge 2 years ago
awan-10
awan-10 approved these changes on 2023-08-29
awan-10 Merge branch 'master' into HeyangQin/enable_hpz_nograd
c4410441
HeyangQin change the way to detect no_grad
355d9d85
HeyangQin Merge branch 'master' into HeyangQin/enable_hpz_nograd
3865dab3
HeyangQin fix format
3769092e
HeyangQin Merge branch 'HeyangQin/enable_hpz_nograd' of https://github.com/micr…
502c2e46
disabled auto-merge 2 years ago
Manually disabled by user
HeyangQin HeyangQin merged 462def45 into master 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone