pytorch-lightning
fix: restore cached AMP step context after no_grad workaround
#21616

Merged

fix: restore cached AMP step context after no_grad workaround #21616

deependujha merged 5 commits into Lightning-AI:master from littlebullGit:fix/21611-autocast-cache-enabled

littlebullGit requested a review from

lantiga 62 days ago

littlebullGit requested a review from

tchaton 62 days ago

littlebullGit requested a review from

justusschock 62 days ago

littlebullGit requested a review from

ethanwharris 62 days ago

github-actions added pl

fix: restore cached AMP step context after no_grad workaround

af1af993

littlebullGit force pushed from 1511c8ab to af1af993 62 days ago

chore: trigger ci

cb2c2620

justusschock approved these changes on 2026-03-30

Merge branch 'master' into fix/21611-autocast-cache-enabled

0a330ffc

chore: trigger ci

3d5a8d63

deependujha requested changes on 2026-03-31

test: add CUDA coverage for AMP no_grad cache handling

167a2444

deependujha approved these changes on 2026-04-01

deependujha merged 4a548c96 into master 57 days ago

Reviewers

deependujha

justusschock

lantiga

tchaton

ethanwharris

Assignees

No one assigned

Labels

Milestone

No milestone