Skip to content

Commit f1e19a7

Browse files
authored
[validations] Perform test_ops only on CUDA binaries (#1657)
1 parent b91724c commit f1e19a7

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.github/scripts/validate_binaries.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ else
4949
export PATH=${OLD_PATH}
5050
fi
5151

52-
if [[ ${INCLUDE_TEST_OPS} == 'true' ]]; then
52+
if [[ ${INCLUDE_TEST_OPS} == 'true' && ${MATRIX_GPU_ARCH_TYPE} == 'cuda' ]]; then
5353
source ./.github/scripts/validate_test_ops.sh
5454
fi
5555

.github/scripts/validate_test_ops.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@ retry git clone ${BRANCH} --depth 1 https://github.com/pytorch/pytorch.git
1818
retry git submodule update --init --recursive
1919
pushd pytorch
2020

21-
pip install expecttest numpy pyyaml jinja2 packaging xmlrunner hypothesis unittest-xml-reporting
21+
pip install expecttest numpy pyyaml jinja2 packaging hypothesis unittest-xml-reporting
2222

2323
# Run test_ops validation
2424
export CUDA_LAUNCH_BLOCKING=1
25-
python3 test/test_ops.py
25+
python3 test/test_ops.py TestCommonCUDA

0 commit comments

Comments
 (0)