pytorch_geometric icon indicating copy to clipboard operation
pytorch_geometric copied to clipboard

Is it possible to get GPU acceleration support on the M1 chip?

Open HuangStomach opened this issue 2 years ago • 2 comments

🚀 The feature, motivation and pitch

You can see from pytorch's blog that the latest version of pytorch already supports GPU acceleration on M1 chips, which can be easily achieved by directly using to('mps'), may I ask if pyg also has plans to do similar support?

Alternatives

No response

Additional context

No response

HuangStomach avatar Jun 07 '22 02:06 HuangStomach

I assume that you can get GPU acceleration on all/most of PyTorch functionality, which should be directly visible when running operators that do not make use of specialized functionality in torch-scatter or torch-sparse, e.g., GCNConv or SAGEConv. For our own external operators, I am afraid that M1 support seems to be of lower priority. We would need to integrate this with the help of Apply or the community.

rusty1s avatar Jun 09 '22 08:06 rusty1s

Not much of a coder, but it be awesome to have pyg support of GPU acceleration!

leo-oneill avatar Aug 10 '22 17:08 leo-oneill