File tree Expand file tree Collapse file tree 1 file changed +13
-2
lines changed
Expand file tree Collapse file tree 1 file changed +13
-2
lines changed Original file line number Diff line number Diff 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\n striping 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 ()
You can’t perform that action at this time.
0 commit comments