search

rapidsai / cugraph

cuGraph - RAPIDS Graph Analytics Library
https://docs.rapids.ai/api/cugraph/stable/
Apache License 2.0
1.68k stars 300 forks source link

Investigate Migrating cugraph.gnn Subpackage #4407

Open alexbarghi-nv opened 4 months ago

alexbarghi-nv commented 4 months ago

cuGraph-PyG and cuGraph-DGL are being decoupled from cuGraph to rely on pylibcugraph instead, but we still have a cugraph.gnn subpackage within cugraph that both libraries need. We should find a better long-term home for that subpackage, especially since we are thinking of changing the release cycle of cuGraph-PyG and cuGraph-DGL and those packages are very tightly coupled with cugraph.gnn.

alexbarghi-nv commented 4 months ago

This is complicated by the distributed sampler's reliance on cuDF - maybe there could be an alternative where pytorch tensors are written instead so that cuDF could be an optional dependency.

alexbarghi-nv commented 2 months ago

Pushing to 24.12 for now.