forked from apache/mxnet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
prepare_mkldnn.sh
executable file
·135 lines (125 loc) · 5.22 KB
/
prepare_mkldnn.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
#!/bin/bash
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
# set -ex
#
# All modification made by Intel Corporation: © 2016 Intel Corporation
#
# All contributions by the University of California:
# Copyright (c) 2014, 2015, The Regents of the University of California (Regents)
# All rights reserved.
#
# All other contributions:
# Copyright (c) 2014, 2015, the respective contributors
# All rights reserved.
# For the list of contributors go to https://github.com/BVLC/caffe/blob/master/CONTRIBUTORS.md
#
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of Intel Corporation nor the names of its contributors
# may be used to endorse or promote products derived from this software
# without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
MXNET_ROOTDIR="$(pwd)"
MKLDNN_ROOTDIR="$MXNET_ROOTDIR/3rdparty/mkldnn/"
MKLDNN_SRCDIR="$MKLDNN_ROOTDIR/src"
MKLDNN_BUILDDIR="$MKLDNN_ROOTDIR/build"
MKLDNN_INSTALLDIR="$MKLDNN_ROOTDIR/install"
MKLDNN_LIBDIR="$MXNET_ROOTDIR/lib"
# MKLDNN install destination
HOME_MKLDNN=$1
if [ ! -z "$HOME_MKLDNN" ]; then
mkdir -p $HOME_MKLDNN
if [ ! -w $HOME_MKLDNN ]; then
echo "MKLDNN install to $HOME_MKLDNN failed, please try with sudo" >&2
exit 1
fi
fi
if [ $(uname) == "Darwin" ]; then
OMP_LIBFILE="$MKLDNN_INSTALLDIR/lib/libiomp5.dylib"
MKLML_LIBFILE="$MKLDNN_INSTALLDIR/lib/libmklml.dylib"
MKLDNN_LIBFILE="$MKLDNN_INSTALLDIR/lib/libmkldnn.0.dylib"
else
OMP_LIBFILE="$MKLDNN_INSTALLDIR/lib/libiomp5.so"
MKLML_LIBFILE="$MKLDNN_INSTALLDIR/lib/libmklml_intel.so"
MKLDNN_LIBFILE="$MKLDNN_INSTALLDIR/lib/libmkldnn.so.0"
fi
if [ -z $MKLDNNROOT ]; then
if [ ! -f $MKLDNN_LIBFILE ]; then
mkdir -p $MKLDNN_INSTALLDIR
cd $MKLDNN_ROOTDIR
if [ -z $MKLROOT ] && [ ! -f $MKLDNN_INSTALLDIR/include/mkl_cblas.h ]; then
rm -rf external && cd scripts && ./prepare_mkl.sh >&2 && cd ..
cp -a external/*/* $MKLDNN_INSTALLDIR/.
fi
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" -DWITH_TEST=OFF -DWITH_EXAMPLE=OFF >&2
NUM_PROC=1
if [[ ! -z $(command -v nproc) ]]; then
NUM_PROC=$(nproc)
elif [[ ! -z $(command -v sysctl) ]]; then
NUM_PROC=$(sysctl -n hw.ncpu)
else
>&2 echo "Can't discover number of cores."
fi
make -C $MKLDNN_BUILDDIR -j${NUM_PROC} VERBOSE=1 >&2
make -C $MKLDNN_BUILDDIR install >&2
rm -rf $MKLDNN_BUILDDIR
mkdir -p $MKLDNN_LIBDIR
cp $OMP_LIBFILE $MKLDNN_LIBDIR
cp $MKLML_LIBFILE $MKLDNN_LIBDIR
cp $MKLDNN_LIBFILE $MKLDNN_LIBDIR
fi
MKLDNNROOT=$MKLDNN_INSTALLDIR
fi
if [ -z $MKLROOT ] && [ -f $MKLDNNROOT/include/mkl_cblas.h ]; then
MKLROOT=$MKLDNNROOT;
fi
# user specified MKLDNN install folder
if [ -d "$HOME_MKLDNN" ]; then
# skip if user specificed MKLDNNROOT
[ "$MKLDNNROOT" != "$HOME_MKLDNN" ] && rsync -a $MKLDNNROOT/include $MKLDNNROOT/lib $HOME_MKLDNN/.
[ "$MKLROOT" != "$HOME_MKLDNN" ] && rsync -a $MKLROOT/include $MKLROOT/lib $HOME_MKLDNN/.
# update ldconfig if possible
if [ -w /etc/ld.so.conf.d ]; then
echo "$HOME_MKLDNN/lib" > /etc/ld.so.conf.d/mxnmkldnn.conf && ldconfig
fi
# return value to calling script (Makefile,cmake)
echo $HOME_MKLDNN $HOME_MKLDNN
else
echo $MKLDNNROOT $MKLROOT
fi