| #!/bin/bash |
| |
| # Required environment variable: $BUILD_ENVIRONMENT |
| # (This is set by default in the Docker images we build, so you don't |
| # need to set it yourself. |
| |
| # shellcheck source=./common.sh |
| source "$(dirname "${BASH_SOURCE[0]}")/common.sh" |
| |
| echo "Testing pytorch (distributed only)" |
| 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 |
| # OMP_NUM_THREADS=2 TORCH_CPP_TEST_MNIST_PATH="test/cpp/api/mnist" build/bin/test_api |
| time python test/run_test.py --verbose -i distributed/test_c10d_common |
| time python test/run_test.py --verbose -i distributed/test_c10d_gloo |
| time python test/run_test.py --verbose -i distributed/test_c10d_nccl |
| time python test/run_test.py --verbose -i distributed/test_c10d_spawn_gloo |
| time python test/run_test.py --verbose -i distributed/test_c10d_spawn_nccl |
| time python test/run_test.py --verbose -i distributed/test_store |
| time python test/run_test.py --verbose -i distributed/test_pg_wrapper |
| time python test/run_test.py --verbose -i distributed/rpc/cuda/test_tensorpipe_agent |
| 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/_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 |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/test_sharded_tensor |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/test_sharded_tensor_reshard |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_chunk |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_elementwise_ops |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_embedding |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_embedding_bag |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_binary_cmp |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_init |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_linear |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_math_ops |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_matrix_ops |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_softmax |
| time python test/run_test.py --verbose -i distributed/_shard/sharded_optim/test_sharded_optim |
| time python test/run_test.py --verbose -i distributed/_shard/test_partial_tensor |
| time python test/run_test.py --verbose -i distributed/_shard/test_replicated_tensor |
| assert_git_not_dirty |
