[BUG] Fix Calculation of # Sampled Nodes/Edges with Zero Input Size #283
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
…into fix-optional-dependencies
…rghi-nv/cugraph-gnn into fix-optional-dependencies
…into fix-zero-input-size
alexbarghi-nv
added
bug
Member
Author
|
/merge |
rapids-bot
bot
merged commit 3f11d45
into
rapidsai:branch-25.10
137 of 139 checks passed
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Fixes a bug found when running
movielens_mnmg.pyon input with a large number of negative edges. Some batches end up with no positive edges of the input type, which causes PyTorch to throw an exception when trying to calculate the number of nodes, since the returned tensor is empty.The fix is to check if the returned tensor is empty, and just set the number of output nodes to zero if it is.