From fbbf3687453aed1b732eee6f6e9050258ce29561 Mon Sep 17 00:00:00 2001 From: Huy Do Date: Thu, 17 Nov 2022 21:33:59 +0000 Subject: [PATCH] Fix distributed test paths when running periodic multigpu job (#89225) Some distributed tests are moved to a new location after https://github.com/pytorch/pytorch/pull/88698. This is currently failing periodic multigpu job: * https://github.com/pytorch/pytorch/actions/runs/3484486207/jobs/5829301159 * https://github.com/pytorch/pytorch/actions/runs/3484486207/jobs/5829301093 Pull Request resolved: https://github.com/pytorch/pytorch/pull/89225 Approved by: https://github.com/clee2000 --- .jenkins/pytorch/multigpu-test.sh | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/.jenkins/pytorch/multigpu-test.sh b/.jenkins/pytorch/multigpu-test.sh index bbd1c370a638..9d7efc969823 100755 --- a/.jenkins/pytorch/multigpu-test.sh +++ b/.jenkins/pytorch/multigpu-test.sh @@ -8,11 +8,6 @@ source "$(dirname "${BASH_SOURCE[0]}")/common.sh" echo "Testing pytorch" -if [ -n "${CI}" ]; then - # TODO move this to docker - # Pin unittest-xml-reporting to freeze printing test summary logic, related: https://github.com/pytorch/pytorch/issues/69014 - pip_install "unittest-xml-reporting<=3.2.0,>=2.0.0" -fi # Disabling tests to see if they solve timeout issues; see https://github.com/pytorch/pytorch/issues/70015 # python tools/download_mnist.py --quiet -d test/cpp/api/mnist @@ -28,8 +23,8 @@ time python test/run_test.py --verbose -i distributed/rpc/cuda/test_tensorpipe_a # FSDP tests for f in test/distributed/fsdp/*.py ; do time python test/run_test.py --verbose -i "${f#*/}" ; done # ShardedTensor tests -time python test/run_test.py --verbose -i distributed/_shard/checkpoint/test_checkpoint -time python test/run_test.py --verbose -i distributed/_shard/checkpoint/test_file_system_checkpoint +time python test/run_test.py --verbose -i distributed/checkpoint/test_checkpoint +time python test/run_test.py --verbose -i distributed/checkpoint/test_file_system_checkpoint time python test/run_test.py --verbose -i distributed/_shard/sharding_spec/test_sharding_spec time python test/run_test.py --verbose -i distributed/_shard/sharding_plan/test_sharding_plan time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/test_megatron_prototype