cugraph icon indicating copy to clipboard operation
cugraph copied to clipboard

[FEA] Migrate All PyTorch-Dependent Code to cuGraph-GNN

Open alexbarghi-nv opened this issue 11 months ago • 3 comments

We continue to have issues with PyTorch in this repository's CI. In addition, the code that depends on PyTorch is really only used by cuGraph-GNN and does not really fit the mission of this repository. As discussed over the past few months, we want to migrate as much GNN code as possible to the cugraph-gnn repository.

There are three key pieces of code affected:

  1. The FeatureStore class which is about to be deprecated (in release 25.02)
  2. The BulkSampler class which is also about to be deprecated (in release 25.02)
  3. The DistSampler class, the replacement for BulkSampler, which is a fundamental piece of our GNN infrastructure.

There is also going to be some additional code in the very near future supporting GNN use cases related to GraphRAG, graph databases, and other frameworks (beyond DGL and PyG). This would also better fit within cugraph-gnn.

We propose creating a new package, pylibcugraphgnn, which will contain the bulk sampling code, as well as any other framework-agnostic code and/or thin wrappers around our C++ code for GNN operations. This package will presumably launch with release 25.04.

alexbarghi-nv avatar Dec 10 '24 16:12 alexbarghi-nv