[validations] Perform test_ops only on CUDA binaries (#1657)

pytorch · Dec 29, 2023 · f1e19a7 · f1e19a7
1 parent b91724c
commit f1e19a7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh
@@ -49,7 +49,7 @@ else
         export PATH=${OLD_PATH}
     fi
 
-    if [[ ${INCLUDE_TEST_OPS} == 'true' ]]; then
+    if [[ ${INCLUDE_TEST_OPS} == 'true' &&  ${MATRIX_GPU_ARCH_TYPE} == 'cuda' ]]; then
         source ./.github/scripts/validate_test_ops.sh
     fi
 

diff --git a/.github/scripts/validate_test_ops.sh b/.github/scripts/validate_test_ops.sh
@@ -18,8 +18,8 @@ retry git clone ${BRANCH} --depth 1 https://github.com/pytorch/pytorch.git
 retry git submodule update --init --recursive
 pushd pytorch
 
-pip install expecttest numpy pyyaml jinja2 packaging xmlrunner hypothesis unittest-xml-reporting
+pip install expecttest numpy pyyaml jinja2 packaging hypothesis unittest-xml-reporting
 
 # Run test_ops validation
 export CUDA_LAUNCH_BLOCKING=1
-python3 test/test_ops.py
+python3 test/test_ops.py TestCommonCUDA