[port] pyg-team/pytorch_geometric

[christian-rauch]

Repository (owner/repo)

pyg-team/pytorch_geometric

Repository URL

https://github.com/pyg-team/pytorch_geometric

What does it do, and why port it?

pytorch_geometric is a PyTorch library for 3D and graph processing

How does it use CUDA?

Has .cu/.cuh CUDA kernels in-tree

Existing ROCm/AMD support?

no

What licence is it under?

MIT License

Build dependencies on other CUDA projects (optional)

pyg-lib, see #21

Before submitting

• This is a CUDA project (not already ROCm/AMD-native), and not a fork of another repo
• I searched open issues and didn't find an existing request for this repo