DeepSpeed
Fix no-grad grad-fn lookup in ZeRO hook counting on PyTorch 2.3 (#7830)
#7841

Merged

Fix no-grad grad-fn lookup in ZeRO hook counting on PyTorch 2.3 (#7830) #7841

tjruwase merged 3 commits into deepspeedai:master from tohtana:tohtana/fix-issue7830-zero3-grad-checkpointing-attrerror

Handle grad-fn lookup failure in ZeRO-3 hook counting (#7830)

c6f868f3

Fix grad-fn lookup in no-grad backward hooks (#7830)

29e7048e

tohtana requested a review from

tjruwase 121 days ago

tohtana requested a review from

loadams 121 days ago

tjruwase approved these changes on 2026-03-01

tjruwase enabled auto-merge (squash) 101 days ago

Merge branch 'master' into tohtana/fix-issue7830-zero3-grad-checkpoin…

7588a873

tjruwase merged bffaf457 into master 101 days ago

Reviewers

tjruwase

loadams

Assignees

No one assigned

Labels

None yet

Milestone

No milestone