-
Notifications
You must be signed in to change notification settings - Fork 3k
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
[GraphBolt] Fix blocks
in minibatch when facing with empty edges in subgraph.
#7413
base: master
Are you sure you want to change the base?
Conversation
To trigger regression tests:
|
relation: gb.CSCFormatBase( | ||
indptr=torch.tensor([0, 1, 2]), indices=torch.tensor([1, 0]) | ||
), | ||
reverse_relation: gb.CSCFormatBase( |
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.
so FusedCSCSamplingGraph.sample_neighbors()
always returns sampled_csc for each edge type even it's empty? Could we mention this in docstring or documentation? probably in sample_neighbors()
?
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.
Yes. Let me add docstring.
python/dgl/graphbolt/minibatch.py
Outdated
@@ -209,6 +209,7 @@ def cast_to_minimum_dtype(v: CSCFormatBase): | |||
), | |||
) | |||
for etype, v in subgraph.sampled_csc.items() | |||
if len(v.indices) > 0 |
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.
homograph could return empty graph as well
Description
related to #7410
TODO:
seeds
still occurs in src_node_type of edges, it will always be passed down.Checklist
Please feel free to remove inapplicable items for your PR.
Changes