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 @@ -299,7 +299,7 @@ if [[ "$PACKAGE_TYPE" == 'libtorch' ]]; then
299299 echo " Checking that XNNPACK is available"
300300 build_and_run_example_cpp check-torch-xnnpack
301301else
302- if [[ " $( uname) " != ' Darwin' || " $PACKAGE_TYPE " != * wheel ]] && [[ " $( uname -m) " != " s390x" && [[ " $( uname -m) " != " ppc64le" ]]; then
302+ if [[ " $( uname) " != ' Darwin' || " $PACKAGE_TYPE " != * wheel ]] && [[ " $( uname -m) " != " s390x" ]] && [[ " $( uname -m) " != " ppc64le" ]]; then
303303 echo " Checking that XNNPACK is available"
304304 pushd /tmp
305305 python -c ' import torch.backends.xnnpack; exit(0 if torch.backends.xnnpack.enabled else 1)'
@@ -320,7 +320,7 @@ if [[ "$OSTYPE" == "msys" ]]; then
320320fi
321321
322322# Test that CUDA builds are setup correctly
323- if [[ " $DESIRED_CUDA " != ' cpu' && " $DESIRED_CUDA " != ' xpu' && " $DESIRED_CUDA " != ' cpu-cxx11-abi' && " $DESIRED_CUDA " != * " rocm" * && " $( uname -m) " != " s390x" && [[ " $( uname -m) " != " ppc64le" ]]; then
323+ if [[ " $DESIRED_CUDA " != ' cpu' && " $DESIRED_CUDA " != ' xpu' && " $DESIRED_CUDA " != ' cpu-cxx11-abi' && " $DESIRED_CUDA " != * " rocm" * && " $( uname -m) " != " s390x" && " $( uname -m) " != " ppc64le" ]]; then
324324 if [[ " $PACKAGE_TYPE " == ' libtorch' ]]; then
325325 build_and_run_example_cpp check-torch-cuda
326326 else
You can’t perform that action at this time.
0 commit comments