This repository was archived by the owner on Nov 17, 2023. It is now read-only.
File tree Expand file tree Collapse file tree 1 file changed +2
-9
lines changed Expand file tree Collapse file tree 1 file changed +2
-9
lines changed Original file line number Diff line number Diff line change @@ -22,10 +22,8 @@ ifeq ($(USE_MKLDNN), 1)
22
22
MXNET_INCLDIR = $(ROOTDIR)/include
23
23
ifeq ($(UNAME_S ) , Darwin)
24
24
MKLDNN_LIBFILE = $(MKLDNNROOT)/lib/libmkldnn.1.dylib
25
- MKLDNN_LIB64FILE = $(MKLDNNROOT)/lib64/libmkldnn.1.dylib
26
25
else
27
26
MKLDNN_LIBFILE = $(MKLDNNROOT)/lib/libmkldnn.so.1
28
- MKLDNN_LIB64FILE = $(MKLDNNROOT)/lib64/libmkldnn.so.1
29
27
endif
30
28
endif
31
29
@@ -35,16 +33,11 @@ mkldnn_build: $(MKLDNN_LIBFILE)
35
33
36
34
$(MKLDNN_LIBFILE ) :
37
35
mkdir -p $(MKLDNNROOT ) /lib
38
- cmake $(MKLDNN_SUBMODDIR ) -DCMAKE_INSTALL_PREFIX=$(MKLDNNROOT ) -B$(MKLDNN_BUILDDIR ) -DMKLDNN_ARCH_OPT_FLAGS=" " -DMKLDNN_BUILD_TESTS=OFF -DMKLDNN_BUILD_EXAMPLES=OFF -DMKLDNN_ENABLE_JIT_PROFILING=OFF
36
+ cmake $(MKLDNN_SUBMODDIR ) -DCMAKE_INSTALL_PREFIX=$(MKLDNNROOT ) -DCMAKE_INSTALL_LIBDIR=lib - B$(MKLDNN_BUILDDIR ) -DMKLDNN_ARCH_OPT_FLAGS=" " -DMKLDNN_BUILD_TESTS=OFF -DMKLDNN_BUILD_EXAMPLES=OFF -DMKLDNN_ENABLE_JIT_PROFILING=OFF
39
37
$(MAKE ) -C $(MKLDNN_BUILDDIR ) VERBOSE=1
40
38
$(MAKE ) -C $(MKLDNN_BUILDDIR ) install
41
39
mkdir -p $(MXNET_LIBDIR )
42
- if [ -f " $( MKLDNN_LIB64FILE) " ]; then \
43
- cp $(MKLDNNROOT ) /lib64/libmkldnn* $(MXNET_LIBDIR ) ; \
44
- cp $(MKLDNNROOT ) /lib64/libmkldnn* $(MKLDNNROOT ) /lib/; \
45
- else \
46
- cp $(MKLDNNROOT ) /lib/libmkldnn* $(MXNET_LIBDIR ) ; \
47
- fi
40
+ cp $(MKLDNN_LIBFILE ) $(MXNET_LIBDIR )
48
41
cp $(MKLDNN_BUILDDIR ) /include/mkldnn_version.h $(MXNET_INCLDIR ) /mkldnn/.
49
42
50
43
mkldnn_clean :
You can’t perform that action at this time.
0 commit comments