From 3bbf8b459bb5881cdbd7ea44e9ae4ffdfecf9e0f Mon Sep 17 00:00:00 2001 From: zhuhong61 Date: Thu, 22 Dec 2022 02:51:56 +0000 Subject: [PATCH] Add manywheel:cpu-cxx11-abi checkup for check_binary.sh --- check_binary.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/check_binary.sh b/check_binary.sh index 51b15dc231..93d392bf1b 100755 --- a/check_binary.sh +++ b/check_binary.sh @@ -51,7 +51,7 @@ else install_root="$(dirname $(which python))/../lib/python${py_dot}/site-packages/torch/" fi -if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != *"rocm"* ]]; then +if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRED_CUDA" != *"rocm"* ]]; then # cu90, cu92, cu100, cu101 if [[ ${#DESIRED_CUDA} -eq 4 ]]; then CUDA_VERSION="${DESIRED_CUDA:2:1}.${DESIRED_CUDA:3:1}" @@ -383,7 +383,7 @@ if [[ "$OSTYPE" == "msys" ]]; then fi # Test that CUDA builds are setup correctly -if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != *"rocm"* ]]; then +if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRED_CUDA" != *"rocm"* ]]; then if [[ "$PACKAGE_TYPE" == 'libtorch' ]]; then build_and_run_example_cpp check-torch-cuda else