Skip to content

Commit

Permalink
[release/2.2][SWDEV-476861] Skip test_aot_inductor distributed UTs (#…
Browse files Browse the repository at this point in the history
…1515)

test_aot_inductor was not enabled in 2.2, however, there are direct
calls to run test_aot_inductor UTs in inductor_distributed.
  • Loading branch information
pragupta authored Aug 1, 2024
1 parent eb433b9 commit 640a5e9
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions .ci/pytorch/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -293,8 +293,11 @@ test_inductor_distributed() {
# Smuggle a few multi-gpu tests here so that we don't have to request another large node
echo "Testing multi_gpu tests in test_torchinductor"
pytest test/inductor/test_torchinductor.py -k test_multi_gpu
pytest test/inductor/test_aot_inductor.py -k test_non_default_cuda_device
pytest test/inductor/test_aot_inductor.py -k test_replicate_on_devices
#test_aot_inductor is not supported in release/2.2.
if [[ "$BUILD_ENVIRONMENT" != *rocm* ]]; then
pytest test/inductor/test_aot_inductor.py -k test_non_default_cuda_device
pytest test/inductor/test_aot_inductor.py -k test_replicate_on_devices
fi
pytest test/distributed/_tensor/test_dtensor_compile.py
pytest test/distributed/tensor/parallel/test_fsdp_2d_parallel.py

Expand Down

0 comments on commit 640a5e9

Please sign in to comment.