Skip to content

GitLab

"src/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "9b4d60799a25409506a73ef4fbaa640a9c4e9ee2"
Unverified Commit 1db71cfe authored by yxy235's avatar yxy235 Committed by GitHub
Browse files

[GraphBolt] Fix MiniBatch blocks. (#6766) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
parent 43fb73db
Hide whitespace changes
Inline Side-by-side
Showing with 12 additions and 2 deletions
python/dgl/graphbolt/minibatch.py
View file @ 1db71cfe
...@@ -395,7 +395,12 @@ class MiniBatch: ...@@ -395,7 +395,12 @@ class MiniBatch:
( (
v.indptr, v.indptr,
v.indices, v.indices,
torch.tensor([]), torch.arange(
0,
v.indptr[-1],
device=v.indptr.device,
dtype=v.indptr.dtype,
),
), ),
) )
for etype, v in subgraph.node_pairs.items() for etype, v in subgraph.node_pairs.items()
...@@ -421,7 +426,12 @@ class MiniBatch: ...@@ -421,7 +426,12 @@ class MiniBatch:
( (
node_pairs.indptr, node_pairs.indptr,
node_pairs.indices, node_pairs.indices,
torch.tensor([]), torch.arange(
0,
node_pairs.indptr[-1],
device=node_pairs.indptr.device,
dtype=node_pairs.indptr.dtype,
),
), ),
) )
num_src_nodes = original_row_node_ids.size(0) num_src_nodes = original_row_node_ids.size(0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment