cugraph icon indicating copy to clipboard operation
cugraph copied to clipboard

Published 20 hours ago • verified publisher icon rapidsai

Distributed Sampling in cuGraph-PyG

Open alexbarghi-nv opened this issue 1 year ago • 0 comments

Distributed sampling in cuGraph-PyG. Also renames the existing API to clarify that it is dask based. Adds a dependency on tensordict for cuGraph-PyG which supports the new TensorDictFeatureStore. Also no longer installs torch-cluster and torch-spline-conv in CI for testing since that results in an ImportError and neither of those packages are needed.

Requires PyG 2.5. Should be merged after #4335

Merge after #4355

Closes #4248 Closes #4249 Closes #3383 Closes #3942 Closes #3836 Closes #4202 Closes #4051 Closes #4326 Closes #4252 Partially addresses #3805

alexbarghi-nv avatar May 01 '24 17:05 alexbarghi-nv