use global communicator

rapidsai · Jul 30, 2024 · 7ba4d89 · 7ba4d89
1 parent ddb95d6
commit 7ba4d89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cugraph-dgl/cugraph_dgl/features.py b/python/cugraph-dgl/cugraph_dgl/features.py
@@ -51,7 +51,7 @@ def __init__(
         location: str(optional, default='cpu')
             The location ('cpu' or 'cuda') where data is stored.
         """
-        self.__wg_comm = wgth.get_local_node_communicator()
+        self.__wg_comm = wgth.get_global_communicator()
 
         if len(tensor.shape) > 2:
             raise ValueError("Only 1-D or 2-D tensors are supported by WholeGraph.")