use int64 to store # edges

rapidsai · Jul 24, 2024 · 2b160bf · 2b160bf
1 parent 4d82ee0
commit 2b160bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cugraph-pyg/cugraph_pyg/data/graph_store.py b/python/cugraph-pyg/cugraph_pyg/data/graph_store.py
@@ -271,7 +271,7 @@ def __get_edgelist(self):
             torch.tensor(
                 [self.__edge_indices[et].shape[1] for et in sorted_keys],
                 device="cuda",
-                dtype=torch.int32,
+                dtype=torch.int64,
             )
         )