setting global_shuffle

rapidsai · Aug 8, 2024 · 63065f5 · 63065f5
1 parent a4ac687
commit 63065f5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_mnmg.py b/python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_mnmg.py
@@ -204,6 +204,7 @@ def run_train(
         directory=train_path,
         shuffle=True,
         drop_last=True,
+        global_shuffle=True,
         **kwargs,
     )
 
@@ -217,6 +218,7 @@ def run_train(
         shuffle=True,
         drop_last=True,
         local_seeds_per_call=80000,
+        global_shuffle=False,
         **kwargs,
     )
 
@@ -229,6 +231,7 @@ def run_train(
         directory=valid_path,
         shuffle=True,
         drop_last=True,
+        global_shuffle=False,
         **kwargs,
     )
 

diff --git a/python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_snmg.py b/python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_snmg.py
@@ -127,6 +127,7 @@ def run_train(
         directory=train_path,
         shuffle=True,
         drop_last=True,
+        global_shuffle=True,
         **kwargs,
     )
 
@@ -140,6 +141,7 @@ def run_train(
         shuffle=True,
         drop_last=True,
         local_seeds_per_call=80000,
+        global_shuffle=False,
         **kwargs,
     )
 
@@ -152,6 +154,7 @@ def run_train(
         directory=valid_path,
         shuffle=True,
         drop_last=True,
+        global_shuffle=False,
         **kwargs,
     )