Merge pull request #12 from alexbarghi-nv/set-distributed

[IMP] Set the Default WG Memory Type to 'distributed' for the MNMG PyG Example (rapidsai/cugraph#4532)
rapidsai · Jul 15, 2024 · d370b0f · d370b0f
2 parents e55b2cd + ecc22bf
commit d370b0f
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()