Skip to content

Commit af30f01

Browse files
committed
Add manywheel:cpu-cxx11-abi checkup for check_binary.sh
1 parent 66c1286 commit af30f01

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

check_binary.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ else
3838
install_root="$(dirname $(which python))/../lib/python${py_dot}/site-packages/torch/"
3939
fi
4040

41-
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != *"rocm"* ]]; then
41+
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRED_CUDA" != *"rocm"* ]]; then
4242
# cu90, cu92, cu100, cu101
4343
if [[ ${#DESIRED_CUDA} -eq 4 ]]; then
4444
CUDA_VERSION="${DESIRED_CUDA:2:1}.${DESIRED_CUDA:3:1}"
@@ -370,7 +370,7 @@ if [[ "$OSTYPE" == "msys" ]]; then
370370
fi
371371

372372
# Test that CUDA builds are setup correctly
373-
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != *"rocm"* ]]; then
373+
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRED_CUDA" != *"rocm"* ]]; then
374374
if [[ "$PACKAGE_TYPE" == 'libtorch' ]]; then
375375
build_and_run_example_cpp check-torch-cuda
376376
else

0 commit comments

Comments
 (0)