Lightning-AI / pytorch-lightning UNCLAIMED

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

0 0 0 Python

Ensure correct device is used for autocast when mps is selected as Fabric accelerator (#20876)

* Make sure MPS is used when chosen as accelerator in Fabric
* Added mps tests to connector and Fabric

---------

Co-authored-by: Haga Device <hagadevice@MacBook-Pro-van-Haga-3.local>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
(cherry picked from commit 14b6c3e24f22aa85dceae48527ab86fdee310973)

Arman Naseri committed 10mo ago

7c1d74eaf60185b77736c0d7776a0a432608107b

Parent: 1c1c110

Committed by Jirka B <j.borovec+github@gmail.com> on 6/19/2025, 10:15:14 AM