From 7f8bd106b47d3e51c55cb9e97ed9af068f5e44ac Mon Sep 17 00:00:00 2001 From: Siyuan Liu Date: Tue, 26 Sep 2023 15:59:11 +0000 Subject: [PATCH] reenable ddp test --- test/run_tests.sh | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/test/run_tests.sh b/test/run_tests.sh index c46fe9ec225..00696144076 100755 --- a/test/run_tests.sh +++ b/test/run_tests.sh @@ -115,7 +115,7 @@ function run_stablehlo_compile { function run_xla_backend_mp { echo "Running XLA backend multiprocessing test: $@" - NCCL_DEBUG=INFO MASTER_ADDR=localhost MASTER_PORT=6000 run_test "$@" + MASTER_ADDR=localhost MASTER_PORT=6000 run_test "$@" } function run_torch_op_tests { @@ -218,8 +218,7 @@ function run_mp_op_tests { run_xla_backend_mp "$CDIR/test_torch_distributed_all_reduce_xla_backend.py" run_xla_backend_mp "$CDIR/test_torch_distributed_multi_all_reduce_xla_backend.py" run_xla_backend_mp "$CDIR/test_torch_distributed_reduce_scatter_xla_backend.py" - # Skip for cuda dev container experiment. - # run_xla_backend_mp "$CDIR/test_ddp.py" + run_xla_backend_mp "$CDIR/test_ddp.py" run_xla_backend_mp "$CDIR/test_fsdp_auto_wrap.py" run_xla_backend_mp "$CDIR/test_torch_distributed_fsdp_meta.py" }