dmlc / dmlc/dgl

[MPS] Metal Performance Shaders (MPS) backend for GPU training acceleration(Mac)

Open
#7,106 0 comments 6 reactions 0 assignees View on GitHub
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.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.