Skip to content

Commit 0ca5ca7

Browse files
USE_MKLDNN 0 in build.sh
1 parent f61dd36 commit 0ca5ca7

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

.ci/pytorch/build.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,11 @@ if [[ "$BUILD_ENVIRONMENT" == *aarch64* ]]; then
9797
export USE_MKLDNN_ACL=1
9898
export ACL_ROOT_DIR=/ComputeLibrary
9999
fi
100+
if [[ "$BUILD_ENVIRONMENT" == *ppc64le* ]]; then
101+
export USE_MKLDNN=0
102+
export USE_MKLDNN_ACL=0
103+
104+
fi
100105

101106
if [[ "$BUILD_ENVIRONMENT" == *libtorch* ]]; then
102107
POSSIBLE_JAVA_HOMES=()

0 commit comments

Comments
 (0)