pytorch-lightning
Fix: `no_grad` with AMP bug
#20921
Merged
Overview
Commits
14
Changes
View On GitHub

Fix: `no_grad` with AMP bug #20921

Borda merged 14 commits into Lightning-AI:master from baskrahmer:fix/no-grad-amp-bug
baskrahmer
baskrahmer Disable cache for torch.autocast in amp
7d45eff9
baskrahmer Add a test
6c8572b2
github-actions github-actions added pl
baskrahmer pre-commit
d18fb086
baskrahmer baskrahmer force pushed from 08508b6f to d18fb086 170 days ago
baskrahmer Merge branch 'master' into fix/no-grad-amp-bug
b9123772
baskrahmer Revert import change
600280bd
baskrahmer Format test
14ae4d88
pre-commit-ci[bot] [pre-commit.ci] auto fixes from pre-commit.com hooks
064caf79
baskrahmer baskrahmer marked this pull request as ready for review 170 days ago
baskrahmer baskrahmer requested a review from lantiga lantiga 170 days ago
baskrahmer baskrahmer requested a review from Borda Borda 170 days ago
baskrahmer baskrahmer requested a review from tchaton tchaton 170 days ago
baskrahmer baskrahmer requested a review from justusschock justusschock 170 days ago
baskrahmer baskrahmer requested a review from ethanwharris ethanwharris 170 days ago
baskrahmer Only test for bf16-mixed
70023a1f
baskrahmer Merge branch 'master' into fix/no-grad-amp-bug
7d616917
Borda
Borda commented on 2025-06-23
Borda
baskrahmer
baskrahmer Implement test to reproduce the issue
b4c44015
baskrahmer Implement fix
2ad5b6b5
pre-commit-ci[bot] [pre-commit.ci] auto fixes from pre-commit.com hooks
2c7f75d8
Borda
baskrahmer Update src/lightning/pytorch/plugins/precision/amp.py
4fa3f749
baskrahmer
Borda
Borda commented on 2025-08-08
github-actions github-actions added has conflicts
Borda Merge branch 'master' into fix/no-grad-amp-bug
39d1847f
github-actions github-actions removed has conflicts
Borda
Borda approved these changes on 2025-09-02
Borda Borda merged 216f9ec9 into master 96 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
pl
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum