DeepSpeed
Fix backward compatibility of torch.amp.custom_fwd for PyTorch < 2.4
#7920

Merged

Fix backward compatibility of torch.amp.custom_fwd for PyTorch < 2.4 #7920

tohtana merged 4 commits into deepspeedai:master from tohtana:tohtata/add-faullback-pt23

add fallback torch.amp.custom_fwd/bwd

52f3d042

tohtana requested a review from

tjruwase 154 days ago

chatgpt-codex-connector commented on 2026-03-24

Handle legacy AMP decorators in ZeRO linear fallback

1bd1e9eb

tohtana requested a review from

loadams 153 days ago

simplify test

fe02b6e1

PKUWZP approved these changes on 2026-03-25

tohtana enabled auto-merge (squash) 152 days ago

Merge branch 'master' into tohtata/add-faullback-pt23

660a4c6d

tohtana merged 138f20d9 into master 152 days ago

Reviewers

PKUWZP

tjruwase

loadams

Assignees

No one assigned

Labels

None yet

Milestone

No milestone