Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Add MPS accelerator support for mixed precision (#21209)
* Update device assignment logic to support 'mps' accelerator * Add tests for MPS accelerator mixed precision device selection * Refactor MPS mixed precision device selection tests for parameterized input * chlog * Apply suggestions Co-authored-by: Nicki Skafte Detlefsen <[email protected]> * apply suggestions * Empty-Commit * Empty Commit * Empty-Commit --------- Co-authored-by: jirka <[email protected]> Co-authored-by: Nicki Skafte Detlefsen <[email protected]> Co-authored-by: Jirka Borovec <[email protected]>
B
Bhimraj Yadav committed
9d40bc851f0acd081d60577fb27fda7bb4db8060
Parent: e088694
Committed by GitHub <[email protected]>
on 9/25/2025, 8:17:31 AM