[port] pyg-team/pyg-lib

[christian-rauch]

Repository (owner/repo)

pyg-team/pyg-lib

Repository URL

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

What does it do, and why port it?

pyg-lib is a PyTorch library with ops for 3D data processing, such as kNN and other clustering approaches. It is a dependency of pytorch_geometric.

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)

No response

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