@ -31,9 +31,17 @@ IF(APPLE)
return ( )
ENDIF ( )
MESSAGE ( STATUS "Set ${MKLDNN_INSTALL_DIR}/lib to runtime path" )
# I n t r o d u c e v a r i a b l e s :
# * C M A K E _ I N S T A L L _ L I B D I R
INCLUDE ( GNUInstallDirs )
SET ( LIBDIR "lib" )
if ( CMAKE_INSTALL_LIBDIR MATCHES ".*lib64$" )
SET ( LIBDIR "lib64" )
endif ( )
MESSAGE ( STATUS "Set ${MKLDNN_INSTALL_DIR}/l${LIBDIR} to runtime path" )
SET ( CMAKE_INSTALL_RPATH_USE_LINK_PATH TRUE )
SET ( CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_RPATH}" "${MKLDNN_INSTALL_DIR}/lib" )
SET ( CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_RPATH}" "${MKLDNN_INSTALL_DIR}/ ${LIBDIR} ")
INCLUDE_DIRECTORIES ( ${ MKLDNN_INC_DIR } ) # F o r M K L D N N c o d e t o i n c l u d e i n t e r n a l h e a d e r s .
@ -58,7 +66,7 @@ ExternalProject_Add(
$ { E X T E R N A L _ P R O J E C T _ L O G _ A R G S }
D E P E N D S $ { M K L D N N _ D E P E N D S }
G I T _ R E P O S I T O R Y " h t t p s : / / g i t h u b . c o m / i n t e l / m k l - d n n . g i t "
G I T _ T A G " 8 30 a 1 0 0 5 9 a 0 1 8 c d 2 6 3 4 d 9 4 1 9 5 1 4 0 c f 2 d 8 7 9 0 a 7 5 a "
G I T _ T A G " 8 63 f f 6 e 7 0 4 2 c e c 7 d 2 e 2 9 8 9 7 f e 9 f 0 8 7 2 e 0 8 8 8 b 0 f c "
P R E F I X $ { M K L D N N _ S O U R C E S _ D I R }
U P D A T E _ C O M M A N D " "
C M A K E _ A R G S - D C M A K E _ C X X _ C O M P I L E R = $ { C M A K E _ C X X _ C O M P I L E R }
@ -79,9 +87,9 @@ ExternalProject_Add(
- D M K L R O O T : P A T H = $ { M K L M L _ R O O T }
)
if ( WIN32 )
SET ( MKLDNN_LIB "${MKLDNN_INSTALL_DIR}/ lib /mkldnn.lib" CACHE FILEPATH "mkldnn library." FORCE )
SET ( MKLDNN_LIB "${MKLDNN_INSTALL_DIR}/ ${LIBDIR} /mkldnn.lib" CACHE FILEPATH "mkldnn library." FORCE )
else ( WIN32 )
SET ( MKLDNN_LIB "${MKLDNN_INSTALL_DIR}/ lib /libmkldnn.so" CACHE FILEPATH "mkldnn library." FORCE )
SET ( MKLDNN_LIB "${MKLDNN_INSTALL_DIR}/ ${LIBDIR} /libmkldnn.so" CACHE FILEPATH "mkldnn library." FORCE )
endif ( WIN32 )
ADD_LIBRARY ( shared_mkldnn SHARED IMPORTED GLOBAL )
@ -101,7 +109,7 @@ ADD_DEPENDENCIES(mkldnn ${MKLDNN_PROJECT})
# c o p y t h e r e a l s o . 0 l i b t o i n s t a l l d i r
# i t c a n b e d i r e c t l y c o n t a i n e d i n w h e e l o r c a p i
if ( WIN32 )
SET ( MKLDNN_SHARED_LIB ${ MKLDNN_INSTALL_DIR } / li b/mkldnn.dll)
SET ( MKLDNN_SHARED_LIB ${ MKLDNN_INSTALL_DIR } / bin /mkldnn.dll)
else ( WIN32 )
SET ( MKLDNN_SHARED_LIB ${ MKLDNN_INSTALL_DIR } /libmkldnn.so.0 )
ADD_CUSTOM_COMMAND ( OUTPUT ${ MKLDNN_SHARED_LIB }