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

Enable MKL-DNN in pip packages #16899

Merged
merged 18 commits into from
Feb 15, 2020
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion make/pip/pip_darwin_cpu.mk
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ USE_LIBJPEG_TURBO = 1
# Please refer to MKL_README.md for details

# whether use MKL-DNN library
USE_MKLDNN = 0
USE_MKLDNN = 1

# MKL ML Library folder, need to be root for /usr/local
# Change to User Home directory for standard user
Expand Down
2 changes: 1 addition & 1 deletion make/pip/pip_linux_cpu.mk
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ USE_OPERATOR_TUNING = 1
USE_LIBJPEG_TURBO = 1

# whether use MKL-DNN library
USE_MKLDNN = 0
USE_MKLDNN = 1


# MKL ML Library for Intel CPU/Xeon Phi
Expand Down
2 changes: 1 addition & 1 deletion make/pip/pip_linux_cu100.mk
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ USE_OPERATOR_TUNING = 1
USE_LIBJPEG_TURBO = 1

# whether use MKL-DNN library
USE_MKLDNN = 0
USE_MKLDNN = 1


# MKL ML Library for Intel CPU/Xeon Phi
Expand Down
2 changes: 1 addition & 1 deletion make/pip/pip_linux_cu101.mk
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ USE_OPERATOR_TUNING = 1
USE_LIBJPEG_TURBO = 1

# whether use MKL-DNN library
USE_MKLDNN = 0
USE_MKLDNN = 1


# MKL ML Library for Intel CPU/Xeon Phi
Expand Down
2 changes: 1 addition & 1 deletion make/pip/pip_linux_cu75.mk
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ USE_OPERATOR_TUNING = 1
USE_LIBJPEG_TURBO = 1

# whether use MKL-DNN library
USE_MKLDNN = 0
USE_MKLDNN = 1


# MKL ML Library for Intel CPU/Xeon Phi
Expand Down
2 changes: 1 addition & 1 deletion make/pip/pip_linux_cu80.mk
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ USE_OPERATOR_TUNING = 1
USE_LIBJPEG_TURBO = 1

# whether use MKL-DNN library
USE_MKLDNN = 0
USE_MKLDNN = 1


# MKL ML Library for Intel CPU/Xeon Phi
Expand Down
2 changes: 1 addition & 1 deletion make/pip/pip_linux_cu90.mk
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ USE_OPERATOR_TUNING = 1
USE_LIBJPEG_TURBO = 1

# whether use MKL-DNN library
USE_MKLDNN = 0
USE_MKLDNN = 1


# MKL ML Library for Intel CPU/Xeon Phi
Expand Down
2 changes: 1 addition & 1 deletion make/pip/pip_linux_cu91.mk
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ USE_OPERATOR_TUNING = 1
USE_LIBJPEG_TURBO = 1

# whether use MKL-DNN library
USE_MKLDNN = 0
USE_MKLDNN = 1


# MKL ML Library for Intel CPU/Xeon Phi
Expand Down
2 changes: 1 addition & 1 deletion make/pip/pip_linux_cu92.mk
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ USE_OPERATOR_TUNING = 1
USE_LIBJPEG_TURBO = 1

# whether use MKL-DNN library
USE_MKLDNN = 0
USE_MKLDNN = 1


# MKL ML Library for Intel CPU/Xeon Phi
Expand Down
5 changes: 1 addition & 4 deletions tools/staticbuild/build_lib.sh
Original file line number Diff line number Diff line change
Expand Up @@ -32,10 +32,7 @@ git submodule update --init --recursive || true
$MAKE DEPS_PATH=$DEPS_PATH DMLCCORE
$MAKE DEPS_PATH=$DEPS_PATH $PWD/3rdparty/tvm/nnvm/lib/libnnvm.a
$MAKE DEPS_PATH=$DEPS_PATH PSLITE

if [[ $VARIANT == *mkl ]]; then
$MAKE DEPS_PATH=$DEPS_PATH mkldnn
fi
$MAKE DEPS_PATH=$DEPS_PATH mkldnn

>&2 echo "Now building mxnet..."
$MAKE DEPS_PATH=$DEPS_PATH
Expand Down