[MPS] Metal Performance Shaders (MPS) backend for GPU training acceleration(Mac)
Open
feature request
- Dominant language
- Python
- Stars
- 14.3k
- Forks
- 3.1k
- PR merge metrics
- No merged PRs in 30d
Description
## 🚀 Feature
## Motivation
[Metal Performance Shaders](https://developer.apple.com/documentation/metalperformanceshaders): Optimize graphics and compute performance with kernels that are fine-tuned for the unique characteristics of each Metal GPU family.
PyTorch supports `mps` device since 1.12: https://pytorch.org/docs/stable/notes/mps.html
Support it in DGL is requested by https://discuss.dgl.ai/t/dgl-with-mps-device/4238.
## Alternatives
## Pitch
## Additional context
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.