From 5cdef4ef7f0df564fd85088c86e0e961958c5ef5 Mon Sep 17 00:00:00 2001 From: Alex Barghi <105237337+alexbarghi-nv@users.noreply.github.com> Date: Mon, 15 Jul 2024 16:48:29 -0400 Subject: [PATCH] [IMP] Set the Default WG Memory Type to 'distributed' for the MNMG PyG Example (#4532) For MNMG processing, the only supported memory type is `distributed`. While it is possible to test using `torchrun` on a single machine, when running on multiple machines, or a SLURM cluster, etc., anything other than `distributed` won't work. Therefore, this PR makes `distributed` the default memory type for that example to avoid user confusion. Authors: - Alex Barghi (https://github.com/alexbarghi-nv) Approvers: - Rick Ratzel (https://github.com/rlratzel) URL: https://github.com/rapidsai/cugraph/pull/4532 --- python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_mnmg.py | 2 +- 1 file changed, 1 insertion(+), 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 index be6447208ce..7002d7ebded 100644 --- 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()