Skip to content

Commit a023195

Browse files
authored
Add manywheel:cpu-cxx11-abi checkup for check_binary.sh (#1251)
1 parent 52f2493 commit a023195

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
@@ -51,7 +51,7 @@ else
5151
install_root="$(dirname $(which python))/../lib/python${py_dot}/site-packages/torch/"
5252
fi
5353

54-
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != *"rocm"* ]]; then
54+
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRED_CUDA" != *"rocm"* ]]; then
5555
# cu90, cu92, cu100, cu101
5656
if [[ ${#DESIRED_CUDA} -eq 4 ]]; then
5757
CUDA_VERSION="${DESIRED_CUDA:2:1}.${DESIRED_CUDA:3:1}"
@@ -383,7 +383,7 @@ if [[ "$OSTYPE" == "msys" ]]; then
383383
fi
384384

385385
# Test that CUDA builds are setup correctly
386-
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != *"rocm"* ]]; then
386+
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRED_CUDA" != *"rocm"* ]]; then
387387
if [[ "$PACKAGE_TYPE" == 'libtorch' ]]; then
388388
build_and_run_example_cpp check-torch-cuda
389389
else

0 commit comments

Comments
 (0)