File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 38
38
install_root=" $( dirname $( which python) ) /../lib/python${py_dot} /site-packages/torch/"
39
39
fi
40
40
41
- if [[ " $DESIRED_CUDA " != ' cpu' && " $DESIRED_CUDA " != * " rocm" * ]]; then
41
+ if [[ " $DESIRED_CUDA " != ' cpu' && " $DESIRED_CUDA " != ' cpu-cxx11-abi ' && " $DESIRED_CUDA " != * " rocm" * ]]; then
42
42
# cu90, cu92, cu100, cu101
43
43
if [[ ${# DESIRED_CUDA} -eq 4 ]]; then
44
44
CUDA_VERSION=" ${DESIRED_CUDA: 2: 1} .${DESIRED_CUDA: 3: 1} "
@@ -370,7 +370,7 @@ if [[ "$OSTYPE" == "msys" ]]; then
370
370
fi
371
371
372
372
# 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
374
374
if [[ " $PACKAGE_TYPE " == ' libtorch' ]]; then
375
375
build_and_run_example_cpp check-torch-cuda
376
376
else
You can’t perform that action at this time.
0 commit comments