diff options
| author | Gu, Jinghui <jinghui.gu@intel.com> | 2018-12-25 22:54:16 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-12-25 22:56:51 -0800 |
| commit | 12e0ed55b4327d41052651ddbae58506c5957ab8 (patch) | |
| tree | 73997a2bc4b3e0c2e8f50f0510544420379431e1 /.jenkins/caffe2 | |
| parent | 2fe5c29d81771de4acbeb692ed775d15139646b7 (diff) | |
| download | pytorch-12e0ed55b4327d41052651ddbae58506c5957ab8.tar.gz pytorch-12e0ed55b4327d41052651ddbae58506c5957ab8.tar.bz2 pytorch-12e0ed55b4327d41052651ddbae58506c5957ab8.zip | |
Upgrade MKL-DNN to version 0.17 and static build MKL-DNN (#15504)
Summary:
Upgrade MKl-DNN to 0.17 and static build MKL-DNN to fix the potentail build error due to old mkldnn version in host system.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15504
Differential Revision: D13547885
Pulled By: soumith
fbshipit-source-id: 46f790a3d9289c1e153e51c62be17c5206ea8f9a
Diffstat (limited to '.jenkins/caffe2')
| -rwxr-xr-x | .jenkins/caffe2/build.sh | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/.jenkins/caffe2/build.sh b/.jenkins/caffe2/build.sh index 8c42ee36ea..4beb22d905 100755 --- a/.jenkins/caffe2/build.sh +++ b/.jenkins/caffe2/build.sh @@ -132,6 +132,7 @@ CMAKE_ARGS+=("-DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX}") if [[ $BUILD_ENVIRONMENT == *mkl* ]]; then CMAKE_ARGS+=("-DBLAS=MKL") + CMAKE_ARGS+=("-DUSE_MKLDNN=ON") fi if [[ $BUILD_ENVIRONMENT == py2-cuda9.0-cudnn7-ubuntu16.04 ]]; then |