fix torch::deploy OSS CI: previous the tests for torch::deploy will be skipped

[ghstack-poisoned]
This commit is contained in:
Shunting Zhang 2021-10-26 14:21:11 -07:00
parent c11e1d7108
commit 9c10c69f79

View File

@ -523,7 +523,7 @@ elif [[ "${BUILD_ENVIRONMENT}" == *libtorch* ]]; then
# TODO: run some C++ tests
echo "no-op at the moment"
elif [[ "${BUILD_ENVIRONMENT}" == *-test1 || "${JOB_BASE_NAME}" == *-test1 || ("${SHARD_NUMBER}" == 1 && $NUM_TEST_SHARDS -gt 1) ]]; then
if [[ "${BUILD_ENVIRONMENT}" == *linux-xenial-cuda11.1*-test1* ]]; then
if [[ "${BUILD_ENVIRONMENT}" == *linux-xenial-cuda11.3*-test1* ]]; then
test_torch_deploy
fi
test_without_numpy