File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 275275if [[ " $PACKAGE_TYPE " == ' libtorch' ]]; then
276276 echo " Checking that MKL is available"
277277 build_and_run_example_cpp check-torch-mkl
278- elif [[ " $( uname -m) " != " arm64" && " $( uname -m) " != " s390x" ]]; then
278+ elif [[ " $( uname -m) " != " arm64" && " $( uname -m) " != " s390x" && " $( uname -m ) " != " ppc64le " ]]; then
279279 if [[ " $( uname) " != ' Darwin' || " $PACKAGE_TYPE " != * wheel ]]; then
280280 if [[ " $( uname -m) " == " aarch64" ]]; then
281281 echo " Checking that MKLDNN is available on aarch64"
@@ -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" ]]; 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" ]]; 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