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 51
51
install_root=" $( dirname $( which python) ) /../lib/python${py_dot} /site-packages/torch/"
52
52
fi
53
53
54
- if [[ " $DESIRED_CUDA " != ' cpu' && " $DESIRED_CUDA " != * " rocm" * ]]; then
54
+ if [[ " $DESIRED_CUDA " != ' cpu' && " $DESIRED_CUDA " != ' cpu-cxx11-abi ' && " $DESIRED_CUDA " != * " rocm" * ]]; then
55
55
# cu90, cu92, cu100, cu101
56
56
if [[ ${# DESIRED_CUDA} -eq 4 ]]; then
57
57
CUDA_VERSION=" ${DESIRED_CUDA: 2: 1} .${DESIRED_CUDA: 3: 1} "
@@ -383,7 +383,7 @@ if [[ "$OSTYPE" == "msys" ]]; then
383
383
fi
384
384
385
385
# 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
387
387
if [[ " $PACKAGE_TYPE " == ' libtorch' ]]; then
388
388
build_and_run_example_cpp check-torch-cuda
389
389
else
You can’t perform that action at this time.
0 commit comments