-
Notifications
You must be signed in to change notification settings - Fork 310
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[FEA] New Graph Interface and Loaders for Distributed Sampling in DGL #4486
[FEA] New Graph Interface and Loaders for Distributed Sampling in DGL #4486
Conversation
…cugraph into dgl-dist-sampling
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Mostly looks good to me, Have added some nits. I think we should add a function to go from dgl.DGLGraph to cugraph_dgl.Graph , having that will help a lot in testing and usability for folks.
Co-authored-by: Vibhu Jawa <[email protected]>
Co-authored-by: Vibhu Jawa <[email protected]>
…cugraph into dgl-dist-sampling
Co-authored-by: Vibhu Jawa <[email protected]>
…cugraph into dgl-dist-sampling
I added |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, thanks for all the changes @alexbarghi-nv , Looks good to go from my side
# Have to use import_optional even though these are required | ||
# dependencies in order to build properly. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
That's interesting, and unfortunate. I'm glad you added this comment.
/merge |
b38e010
into
rapidsai:branch-24.08
See rapidsaigh-4486 for adding torchdata/pydantic and rapidsaigh-4464 for introducing the torch pin due to NCCL difficulties. It is entirely possible that this still fails due to the NCCL issue!
DGLGraph
FeatureStorage
This PR also adds
pydantic
andtorchdata
as test dependencies of cuGraph-DGL, because they are unlisted runtime dependencies of the latest version of DGL (which we will be testing with in the newcugraph-gnn
repository).This PR also changes the test file structure to match the rest of cuGraph, by using the _mg suffix to denote multi-GPU tests and moving them to the main directory under "tests".
New examples will be added in a future PR. This PR is just for basic functionality and tests.
Closes #3805
Closes rapidsai/cugraph-gnn#4
Closes #4400
Closes #4247
Closes #3303