Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "6f9ae8d61eca4a2841bb06b47a993c523de6f43c"
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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
python/dgl/graphbolt/impl/fused_csc_sampling_graph.py
View file @ e9162491
...@@ -417,10 +417,8 @@ class FusedCSCSamplingGraph(SamplingGraph): ...@@ -417,10 +417,8 @@ class FusedCSCSamplingGraph(SamplingGraph):
and ORIGINAL_EDGE_ID in self.edge_attributes and ORIGINAL_EDGE_ID in self.edge_attributes
) )
if has_original_eids: if has_original_eids:
original_edge_ids = torch.index_select( original_edge_ids = torch.ops.graphbolt.index_select(
self.edge_attributes[ORIGINAL_EDGE_ID], self.edge_attributes[ORIGINAL_EDGE_ID], original_edge_ids
dim=0,
index=original_edge_ids,
) )
if type_per_edge is None: if type_per_edge is None:
# The sampled graph is already a homogeneous graph. # 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