From 34a1ecd4034bf61c911202f3b1babca358f0119b Mon Sep 17 00:00:00 2001 From: Alexandria Barghi Date: Thu, 1 Aug 2024 09:04:00 -0700 Subject: [PATCH] patch --- python/cugraph-pyg/cugraph_pyg/data/feature_store.py | 2 +- python/cugraph-pyg/cugraph_pyg/data/graph_store.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/cugraph-pyg/cugraph_pyg/data/feature_store.py b/python/cugraph-pyg/cugraph_pyg/data/feature_store.py index 0adef9f9135..e7aa461e390 100644 --- a/python/cugraph-pyg/cugraph_pyg/data/feature_store.py +++ b/python/cugraph-pyg/cugraph_pyg/data/feature_store.py @@ -157,7 +157,7 @@ def __init__(self, memory_type="distributed", location="cpu"): self.__features = {} - self.__wg_comm = wgth.get_local_node_communicator() + self.__wg_comm = wgth.get_get_global_communicator() self.__wg_type = memory_type self.__wg_location = location diff --git a/python/cugraph-pyg/cugraph_pyg/data/graph_store.py b/python/cugraph-pyg/cugraph_pyg/data/graph_store.py index 01af7fd6ed0..03a409a6489 100644 --- a/python/cugraph-pyg/cugraph_pyg/data/graph_store.py +++ b/python/cugraph-pyg/cugraph_pyg/data/graph_store.py @@ -260,7 +260,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, ) )