update code

rapidsai · Jul 15, 2024 · ecc22bf · ecc22bf
1 parent 8e6619c
commit ecc22bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_mnmg.py b/python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_mnmg.py
@@ -345,7 +345,7 @@ def parse_args():
     parser.add_argument("--dataset_root", type=str, default="dataset")
     parser.add_argument("--dataset", type=str, default="ogbn-products")
     parser.add_argument("--skip_partition", action="store_true")
-    parser.add_argument("--wg_mem_type", type=str, default="chunked")
+    parser.add_argument("--wg_mem_type", type=str, default="distributed")
 
     return parser.parse_args()