Fix build scripts for MKL (#9698)

This commit is contained in:
Shanqing Cai 2017-05-05 18:16:04 -04:00 committed by Vijay Vasudevan
parent 65f8c5ca31
commit cae8ed1ca5
3 changed files with 3 additions and 8 deletions

1
.gitignore vendored
View File

@ -4,6 +4,7 @@ node_modules
/.bazelrc
/.tf_configure.bazelrc
/bazel-*
/bazel_pip
/third_party/eigen3/mkl_include
/third_party/mkl/*
/third_party/py/numpy/numpy_include

4
configure vendored
View File

@ -210,8 +210,8 @@ if [ "$TF_NEED_MKL" == "1" ]; then # TF_NEED_MKL
ARCHIVE_BASENAME=mklml_lnx_2018.0.20170425.tgz
GITHUB_RELEASE_TAG=v0.7
MKLURL="https://github.com/01org/mkl-dnn/releases/download/$GITHUB_RELEASE_TAG/$ARCHIVE_BASENAME"
if ! [ -e "$DST/third_party/mkl/$ARCHIVE_BASENAME" ]; then
wget --no-check-certificate -P $DST/third_party/mkl/ $MKLURL
if ! [ -e "${DST}/third_party/mkl/${ARCHIVE_BASENAME}" ]; then
curl -fSsL -o "${DST}/third_party/mkl/${ARCHIVE_BASENAME}" "${MKLURL}"
fi
tar -xzf $DST/third_party/mkl/$ARCHIVE_BASENAME -C $DST/third_party/mkl/
extracted_dir_name="${ARCHIVE_BASENAME%.*}"

View File

@ -28,12 +28,6 @@ set -e
CONTAINER_TYPE=$( echo "$1" | tr '[:upper:]' '[:lower:]' )
shift 1
# Enable support for MKL, for Linux only.
if [[ $(uname) == "Linux" ]]; then
export TF_NEED_MKL=1
export TF_DOWNLOAD_MKL=1
fi
# Enable support for Google Cloud Platform (GCP)
export TF_NEED_GCP=1
# Enable support for HDFS