Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
upgrade MKL (#8378)
Browse files Browse the repository at this point in the history
  • Loading branch information
szha authored and piiswrong committed Oct 26, 2017
1 parent 1df79b4 commit 840b9f1
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions prepare_mkl.sh
Original file line number Diff line number Diff line change
Expand Up @@ -75,14 +75,14 @@ MXNET_ROOT=`dirname $0`
USE_MKLML=0
# NOTE: if you update the following line, please also update the dockerfile at
# tests/ci_build/Dockerfile.mkl
VERSION_MATCH=20170720
VERSION_MATCH=20170908
PLATFORM=$(uname)
if [ $PLATFORM == "Darwin" ]; then
INFIX=mac
elif [ $PLATFORM == "Linux" ]; then
INFIX=lnx
fi
ARCHIVE_BASENAME=mklml_${INFIX}_2018.0.20170720.tgz
ARCHIVE_BASENAME=mklml_${INFIX}_2018.0.20170908.tgz
MKL_CONTENT_DIR=`echo $ARCHIVE_BASENAME | rev | cut -d "." -f 2- | rev`
MKLURL="https://github.com/01org/mkl-dnn/releases/download/v0.10/$ARCHIVE_BASENAME"
# there are diffrent MKL lib to be used for GCC and for ICC
Expand Down
2 changes: 1 addition & 1 deletion tests/ci_build/Dockerfile.mklml_gpu
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ RUN /install/ubuntu_install_python.sh
COPY install/ubuntu_install_scala.sh /install/
RUN /install/ubuntu_install_scala.sh

RUN wget --no-check-certificate -O /tmp/mklml.tgz https://github.com/01org/mkl-dnn/releases/download/v0.10/mklml_lnx_2018.0.20170720.tgz
RUN wget --no-check-certificate -O /tmp/mklml.tgz https://github.com/01org/mkl-dnn/releases/download/v0.10/mklml_lnx_2018.0.20170908.tgz
RUN tar -zxvf /tmp/mklml.tgz && cp -rf mklml_*/* /usr/local/ && rm -rf mklml_*

ENV LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/lib

0 comments on commit 840b9f1

Please sign in to comment.