Skip to content

Commit

Permalink
[IMP] Set the Default WG Memory Type to 'distributed' for the MNMG Py…
Browse files Browse the repository at this point in the history
…G 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: #4532
  • Loading branch information
alexbarghi-nv authored Jul 15, 2024
1 parent b55c279 commit 5cdef4e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/cugraph-pyg/cugraph_pyg/examples/gcn_dist_mnmg.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand Down

0 comments on commit 5cdef4e

Please sign in to comment.