pytorch-lightning
9d40bc85 - Add MPS accelerator support for mixed precision (#21209)

Commit
86 days ago
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 <skaftenicki@gmail.com> * apply suggestions * Empty-Commit * Empty Commit * Empty-Commit --------- Co-authored-by: jirka <jirka.borovec@seznam.cz> Co-authored-by: Nicki Skafte Detlefsen <skaftenicki@gmail.com> Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Author
Parents
Loading