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

[MXNET-367] update mkldnn to v0.14 and disable building test examples #10736

Merged
merged 4 commits into from
May 4, 2018
Merged
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
Next Next commit
update mkldnn to v0.14 and disable building test examples
  • Loading branch information
ashokei committed Apr 28, 2018
commit a2622980f527b70cc0ff42b1271350fc80fd73b1
4 changes: 3 additions & 1 deletion prepare_mkldnn.sh
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,7 @@

MXNET_ROOTDIR="$(pwd)"
MKLDNN_ROOTDIR="$MXNET_ROOTDIR/3rdparty/mkldnn/"
MKLDNN_VERSION="v0.14"
MKLDNN_SRCDIR="$MKLDNN_ROOTDIR/src"
MKLDNN_BUILDDIR="$MKLDNN_ROOTDIR/build"
MKLDNN_INSTALLDIR="$MKLDNN_ROOTDIR/install"
Expand Down Expand Up @@ -93,7 +94,8 @@ if [ ! -f $MKLDNN_LIBFILE ]; then
echo "Building MKLDNN ..." >&2
cd $MXNET_ROOTDIR
g++ --version >&2
cmake $MKLDNN_ROOTDIR -DCMAKE_INSTALL_PREFIX=$MKLDNN_INSTALLDIR -B$MKLDNN_BUILDDIR -DARCH_OPT_FLAGS="-mtune=generic" >&2
git reset --hard $MKLDNN_VERSION >&2
cmake $MKLDNN_ROOTDIR -DCMAKE_INSTALL_PREFIX=$MKLDNN_INSTALLDIR -B$MKLDNN_BUILDDIR -DARCH_OPT_FLAGS="-mtune=generic" -DWITH_TEST=OFF -DWITH_EXAMPLE=OFF >&2
NUM_PROC=1
if [[ ! -z $(command -v nproc) ]]; then
NUM_PROC=$(nproc)
Expand Down