heat icon indicating copy to clipboard operation
heat copied to clipboard

Support Apple's MPS backend

Open ClaudiaComito opened this issue 2 years ago • 6 comments

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 avatar Nov 26 '22 06:11 ClaudiaComito

First few tests (PyTorch 1.13):

  • only single precision supported
  • multi-GPU setup unclear
  • PyTorch support does not cover all operations

ClaudiaComito avatar Nov 26 '22 07:11 ClaudiaComito

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

github-actions[bot] avatar Mar 04 '24 02:03 github-actions[bot]

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

github-actions[bot] avatar Jun 24 '24 02:06 github-actions[bot]

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

github-actions[bot] avatar Aug 26 '24 02:08 github-actions[bot]

In progress #1129

ClaudiaComito avatar Aug 26 '24 04:08 ClaudiaComito