Skip to content

Commit 99e3d1d

Browse files
committed
update
1 parent 7c5362a commit 99e3d1d

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

cmake/inference_lib.cmake

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,12 @@ function(copy_part_of_thrid_party TARGET DST)
7272
copy(${TARGET}
7373
SRCS ${MKLML_LIB} ${MKLML_IOMP_LIB} ${MKLML_INC_DIR}
7474
DSTS ${dst_dir}/lib ${dst_dir}/lib ${dst_dir})
75+
if(WITH_STRIP)
76+
add_custom_command(TARGET ${TARGET} POST_BUILD
77+
COMMAND strip -s ${dst_dir}/lib/libiomp5.so
78+
COMMAND strip -s ${dst_dir}/lib/libmklml_intel.so
79+
COMMENT "striping libiomp5.so\nstriping libmklml_intel.so")
80+
endif()
7581
endif()
7682
elseif(${CBLAS_PROVIDER} STREQUAL EXTERN_OPENBLAS)
7783
set(dst_dir "${DST}/third_party/install/openblas")
@@ -96,9 +102,14 @@ function(copy_part_of_thrid_party TARGET DST)
96102
copy(${TARGET}
97103
SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB}
98104
DSTS ${dst_dir} ${dst_dir}/lib)
105+
if(WITH_STRIP)
106+
add_custom_command(TARGET ${TARGET} POST_BUILD
107+
COMMAND strip -s ${dst_dir}/lib/libmkldnn.so.0
108+
COMMENT "striping libmkldnn.so.0")
109+
endif()
99110
add_custom_command(TARGET ${TARGET} POST_BUILD
100-
COMMAND ${CMAKE_COMMAND} -E create_symlink ${dst_dir}/lib/libmkldnn.so.0 ${dst_dir}/lib/libdnnl.so.1
101-
COMMAND ${CMAKE_COMMAND} -E create_symlink ${dst_dir}/lib/libmkldnn.so.0 ${dst_dir}/lib/libdnnl.so.2
111+
COMMAND ${CMAKE_COMMAND} -E create_symlink libmkldnn.so.0 ${dst_dir}/lib/libdnnl.so.1
112+
COMMAND ${CMAKE_COMMAND} -E create_symlink libmkldnn.so.0 ${dst_dir}/lib/libdnnl.so.2
102113
COMMENT "Make a symbol link of libmkldnn.so.0")
103114
endif()
104115
endif()

0 commit comments

Comments
 (0)