Fix build scripts for MKL (#9698)
This commit is contained in:
parent
65f8c5ca31
commit
cae8ed1ca5
1
.gitignore
vendored
1
.gitignore
vendored
@ -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
4
configure
vendored
@ -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%.*}"
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user