Skip to content

Commit

Permalink
Re-enable torchdynamo unit tests
Browse files Browse the repository at this point in the history
ghstack-source-id: cb1be51e6a02d605b8b6be0eeaa159dbd4974b55
Pull Request resolved: #86658
  • Loading branch information
desertfire committed Oct 11, 2022
1 parent 937d677 commit 59e6dca
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions .jenkins/pytorch/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -651,7 +651,7 @@ test_vec256() {

test_dynamo() {
pushd ../torchdynamo
pytest test
pytest test/dynamo
popd
}

Expand Down Expand Up @@ -693,8 +693,7 @@ elif [[ "${TEST_CONFIG}" == *dynamo* && "${SHARD_NUMBER}" == 2 && $NUM_TEST_SHAR
install_torchvision
checkout_install_torchdynamo
test_dynamo_shard 2
# Skip running test in the dynamo repo to unblock the dynamo pin update
# test_dynamo
test_dynamo
elif [[ "${SHARD_NUMBER}" == 1 && $NUM_TEST_SHARDS -gt 1 ]]; then
test_without_numpy
install_torchvision
Expand Down

0 comments on commit 59e6dca

Please sign in to comment.