Skip to content

GitLab

Unverified Commit f81e514d authored by Muhammed Fatih BALIN's avatar Muhammed Fatih BALIN Committed by GitHub
Browse files

[GraphBolt][Bug] Quick fix for GPU sampling case. (#6856)

parent 46a566af
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
python/dgl/graphbolt/impl/fused_csc_sampling_graph.py
View file @ f81e514d
...@@ -448,7 +448,9 @@ class FusedCSCSamplingGraph(SamplingGraph): ...@@ -448,7 +448,9 @@ class FusedCSCSamplingGraph(SamplingGraph):
homogeneous_timestamps = [] homogeneous_timestamps = []
for ntype, ids in nodes.items(): for ntype, ids in nodes.items():
ntype_id = self.node_type_to_id[ntype] ntype_id = self.node_type_to_id[ntype]
homogeneous_nodes.append(ids + self.node_type_offset[ntype_id]) homogeneous_nodes.append(
ids + self.node_type_offset[ntype_id].item()
)
if timestamps is not None: if timestamps is not None:
homogeneous_timestamps.append(timestamps[ntype]) homogeneous_timestamps.append(timestamps[ntype])
if timestamps is not None: if timestamps is not None:
......
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