From: Junjie Bai <bai@in.tum.de>
Date: Mon, 8 Apr 2019 20:09:11 +0000 (-0700)
Subject: Turn on mkldnn in most builds except rocm
X-Git-Tag: accepted/tizen/6.5/unified/20211028.231830~336
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=239de1623d07b3b88900ba54a37d43d66e3bffa5;p=platform%2Fupstream%2Fpytorch.git

Turn on mkldnn in most builds except rocm

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18965

Differential Revision: D14836931

Pulled By: bddppq

fbshipit-source-id: 463a9bc5043a1f3194158f7bbfae3b71c6cd4b20
---

diff --git a/.jenkins/pytorch/build.sh b/.jenkins/pytorch/build.sh
index 4e412f7..7420b0b 100755
--- a/.jenkins/pytorch/build.sh
+++ b/.jenkins/pytorch/build.sh
@@ -48,7 +48,10 @@ pip install -q -r requirements.txt || true
 
 # TODO: Don't install this here
 if ! which conda; then
-  if [[ "$BUILD_ENVIRONMENT" == *trusty-py3.6-gcc7.2* ]] || [[ "$BUILD_ENVIRONMENT" == *trusty-py3.6-gcc4.8* ]]; then
+  # In ROCm CIs, we are doing cross compilation on build machines with
+  # intel cpu and later run tests on machines with amd cpu.
+  # Also leave out two builds to make sure non-mkldnn builds still work.
+  if [[ "$BUILD_ENVIRONMENT" != *rocm* && "$BUILD_ENVIRONMENT" != *-trusty-py3.5-* && "$BUILD_ENVIRONMENT" != *-xenial-cuda8-cudnn7-py3-* ]]; then
     pip install -q mkl mkl-devel
     export USE_MKLDNN=1
   else