diff --git a/examples/mc_multigpu_test.py b/examples/mc_multigpu_test.py index 3caed7a..ed72629 100644 --- a/examples/mc_multigpu_test.py +++ b/examples/mc_multigpu_test.py @@ -10,8 +10,8 @@ ) -backend = "nccl" -# backend = "gloo" +# backend = "nccl" +backend = "gloo" # set_seed(42) setup(backend=backend) device = get_device()