pytorch-lightning
Add MPS accelerator support for mixed precision
#21209
Open
Overview
Commits
11
Changes
View On GitHub

Add MPS accelerator support for mixed precision #21209

bhimrazy wants to merge 11 commits into Lightning-AI:master from bhimrazy:feature/mps-amp-support
bhimrazy
bhimrazy Update device assignment logic to support 'mps' accelerator
92f01038
bhimrazy Add tests for MPS accelerator mixed precision device selection
39e7682b
bhimrazy bhimrazy requested a review from lantiga lantiga 10 days ago
bhimrazy bhimrazy requested a review from Borda Borda 10 days ago
bhimrazy bhimrazy requested a review from tchaton tchaton 10 days ago
bhimrazy bhimrazy requested a review from justusschock justusschock 10 days ago
bhimrazy bhimrazy requested a review from ethanwharris ethanwharris 10 days ago
github-actions github-actions added pl
bhimrazy Refactor MPS mixed precision device selection tests for parameterized…
b2d30f1b
bhimrazy bhimrazy assigned bhimrazy bhimrazy 10 days ago
bhimrazy bhimrazy added accelerator: mps
Borda
Borda approved these changes on 2025-09-14
Borda chlog
e7ba39c5
SkafteNicki
SkafteNicki commented on 2025-09-15
bhimrazy Apply suggestions
4db4bc4c
bhimrazy apply suggestions
cf0960f4
Borda Merge branch 'master' into feature/mps-amp-support
8fb3a20c
bhimrazy Merge branch 'master' into feature/mps-amp-support
8918ff5d
SkafteNicki Merge branch 'master' into feature/mps-amp-support
ffa1c010
SkafteNicki
SkafteNicki approved these changes on 2025-09-16
Borda Empty-Commit
c4c72e84
bhimrazy Empty Commit
a79cd252

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
accelerator: mps pl
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum