"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "66394bf6c798a93a1e9536dac4999f77b690174c"
Unverified Commit e9162491 authored by Muhammed Fatih BALIN's avatar Muhammed Fatih BALIN Committed by GitHub
Browse files

[GraphBolt][CUDA] Fix pinned `edge_id` index_select (#6898)

parent 29c3b06d
......@@ -417,10 +417,8 @@ class FusedCSCSamplingGraph(SamplingGraph):
and ORIGINAL_EDGE_ID in self.edge_attributes
)
if has_original_eids:
original_edge_ids = torch.index_select(
self.edge_attributes[ORIGINAL_EDGE_ID],
dim=0,
index=original_edge_ids,
original_edge_ids = torch.ops.graphbolt.index_select(
self.edge_attributes[ORIGINAL_EDGE_ID], original_edge_ids
)
if type_per_edge is None:
# The sampled graph is already a homogeneous graph.
......
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