search

helmholtz-analytics / heat

Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
https://heat.readthedocs.io/
MIT License
209 stars 53 forks source link

Support Apple's MPS backend #1053

Open ClaudiaComito opened 1 year ago

ClaudiaComito commented 1 year ago

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.

ClaudiaComito commented 1 year ago

First few tests (PyTorch 1.13):

github-actions[bot] commented 6 months ago

This issue is stale because it has been open for 60 days with no activity.

github-actions[bot] commented 5 months ago

Branch 1053-Support_Apple_s_MPS_backend created!

github-actions[bot] commented 2 months ago

This issue is stale because it has been open for 60 days with no activity.

github-actions[bot] commented 3 weeks ago

This issue was closed because it has been inactive for 60 days since being marked as stale.

ClaudiaComito commented 3 weeks ago

In progress #1129