rusty1s/pytorch_cluster

Support for AMD GPUs (ROCm) via HIP?

maharjun opened this issue · 2 comments

Is there any way to use pytorch_cluster with AMD GPUs? How feasible is it to implement a HIPified version of pytorch_cluster? has anyone tried it?

Can you take a look at https://github.com/Looong01/pyg-rocm-build and see if that is sufficient for your use-case?

This issue had no activity for 6 months. It will be closed in 2 weeks unless there is some new activity. Is this issue already resolved?