Skip to content

Commit a67486c

Browse files
nightly build changes
1 parent 8827de2 commit a67486c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.ci/pytorch/check_binary.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -275,7 +275,7 @@ fi
275275
if [[ "$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
301301
else
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
320320
fi
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

0 commit comments

Comments
 (0)