heat
heat copied to clipboard
Support Apple's MPS backend
Since version 1.12, PyTorch supports GPU acceleration via Apple's Metal Performance Shaders (MPS) backend. https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/
Feature functionality
device="gpu"
should be a valid argument for ht.factories
if the MPS backend is available for torch versions >= 1.12.
Ongoing research on branch features/1053-support-Apple-silicon-GPUs
, will document here.
First few tests (PyTorch 1.13):
- only single precision supported
- multi-GPU setup unclear
- PyTorch support does not cover all operations
This issue is stale because it has been open for 60 days with no activity.
Branch 1053-Support_Apple_s_MPS_backend created!
This issue is stale because it has been open for 60 days with no activity.
This issue was closed because it has been inactive for 60 days since being marked as stale.
In progress #1129