mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
remove references to ort_test
Fixes #ISSUE_NUMBER ort_test, -test1, -test2 is from before migration to GHA? removing dead/no longer relevant code? Pull Request resolved: https://github.com/pytorch/pytorch/pull/76091 Approved by: https://github.com/janeyx99
This commit is contained in:
parent
75be4f9cdb
commit
4bf5380ec7
|
|
@ -45,8 +45,8 @@ fi
|
|||
################################################################################
|
||||
# C++ tests #
|
||||
################################################################################
|
||||
# Don't run cpp tests a second time in the sharded ort_test2 job
|
||||
if [[ "$BUILD_ENVIRONMENT" != *ort_test2* ]]; then
|
||||
# Only run cpp tests in the first shard, don't run cpp tests a second time in the second shard
|
||||
if [[ "${SHARD_NUMBER:-1}" == "1" ]]; then
|
||||
echo "Running C++ tests.."
|
||||
for test in $(find "$cpp_test_dir" -executable -type f); do
|
||||
case "$test" in
|
||||
|
|
|
|||
|
|
@ -564,12 +564,12 @@ elif [[ "${BUILD_ENVIRONMENT}" == *jit_legacy-test || "${JOB_BASE_NAME}" == *jit
|
|||
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
|
||||
elif [[ "${SHARD_NUMBER}" == 1 && $NUM_TEST_SHARDS -gt 1 ]]; then
|
||||
test_without_numpy
|
||||
install_torchvision
|
||||
test_python_shard 1
|
||||
test_aten
|
||||
elif [[ "${BUILD_ENVIRONMENT}" == *-test2 || "${JOB_BASE_NAME}" == *-test2 || ("${SHARD_NUMBER}" == 2 && $NUM_TEST_SHARDS -gt 1) ]]; then
|
||||
elif [[ "${SHARD_NUMBER}" == 2 && $NUM_TEST_SHARDS -gt 1 ]]; then
|
||||
install_torchvision
|
||||
test_python_shard 2
|
||||
test_libtorch
|
||||
|
|
|
|||
|
|
@ -65,13 +65,13 @@ run_tests() {
|
|||
"$SCRIPT_HELPERS_DIR"/test_custom_backend.bat
|
||||
"$SCRIPT_HELPERS_DIR"/test_libtorch.bat
|
||||
else
|
||||
if [[ "${JOB_BASE_NAME}" == *-test1 || ("${SHARD_NUMBER}" == 1 && $NUM_TEST_SHARDS -gt 1) ]]; then
|
||||
if [[ "${SHARD_NUMBER}" == 1 && $NUM_TEST_SHARDS -gt 1 ]]; then
|
||||
"$SCRIPT_HELPERS_DIR"/test_python_first_shard.bat
|
||||
"$SCRIPT_HELPERS_DIR"/test_libtorch.bat
|
||||
if [[ "${USE_CUDA}" == "1" ]]; then
|
||||
"$SCRIPT_HELPERS_DIR"/test_python_jit_legacy.bat
|
||||
fi
|
||||
elif [[ "${JOB_BASE_NAME}" == *-test2 || ("${SHARD_NUMBER}" == 2 && $NUM_TEST_SHARDS -gt 1) ]]; then
|
||||
elif [[ "${SHARD_NUMBER}" == 2 && $NUM_TEST_SHARDS -gt 1 ]]; then
|
||||
"$SCRIPT_HELPERS_DIR"/test_python_second_shard.bat
|
||||
"$SCRIPT_HELPERS_DIR"/test_custom_backend.bat
|
||||
"$SCRIPT_HELPERS_DIR"/test_custom_script_ops.bat
|
||||
|
|
|
|||
|
|
@ -51,7 +51,7 @@ fi
|
|||
|
||||
# onnxruntime only support py3
|
||||
# "Python.h" not found in py2, needed by TorchScript custom op compilation.
|
||||
if [[ "$BUILD_ENVIRONMENT" == *ort_test1* || "${SHARD_NUMBER}" == "1" ]]; then
|
||||
if [[ "${SHARD_NUMBER}" == "1" ]]; then
|
||||
# These exclusions are for tests that take a long time / a lot of GPU
|
||||
# memory to run; they should be passing (and you will test them if you
|
||||
# run them locally
|
||||
|
|
@ -79,7 +79,7 @@ if [[ "$BUILD_ENVIRONMENT" == *ort_test1* || "${SHARD_NUMBER}" == "1" ]]; then
|
|||
"$top_dir/test/onnx/test_onnx_export.py"
|
||||
fi
|
||||
|
||||
if [[ "$BUILD_ENVIRONMENT" == *ort_test2* || "${SHARD_NUMBER}" == "2" ]]; then
|
||||
if [[ "${SHARD_NUMBER}" == "2" ]]; then
|
||||
# Update the loop for new opsets
|
||||
for i in $(seq 10 15); do
|
||||
pytest "${args[@]}" \
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user