diff options
| author | Junjie Bai <bai@in.tum.de> | 2019-04-08 13:09:11 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-04-08 13:19:14 -0700 |
| commit | 239de1623d07b3b88900ba54a37d43d66e3bffa5 (patch) | |
| tree | ed3aff6abfa8ede8845ae4e742c3e88c6a66a642 | |
| parent | 9b76f69cd3dfd66b4a03f1bfc26cead0887c9cfe (diff) | |
| download | pytorch-239de1623d07b3b88900ba54a37d43d66e3bffa5.tar.gz pytorch-239de1623d07b3b88900ba54a37d43d66e3bffa5.tar.bz2 pytorch-239de1623d07b3b88900ba54a37d43d66e3bffa5.zip | |
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
| -rwxr-xr-x | .jenkins/pytorch/build.sh | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/.jenkins/pytorch/build.sh b/.jenkins/pytorch/build.sh index 4e412f7da5..7420b0bd0d 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 |