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
......@@ -395,7 +395,12 @@ class MiniBatch:
(
v.indptr,
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()
......@@ -421,7 +426,12 @@ class MiniBatch:
(
node_pairs.indptr,
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)
......
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