|
|
|
@ -41,6 +41,7 @@ INCLUDE_DIRECTORIES(${MKLDNN_INCLUDE_DIR})
|
|
|
|
|
IF(${CBLAS_PROVIDER} STREQUAL "MKLML")
|
|
|
|
|
SET(MKLDNN_DEPENDS ${MKLML_PROJECT})
|
|
|
|
|
SET(MKLDNN_MKLROOT ${MKLML_ROOT})
|
|
|
|
|
SET(MKLDNN_IOMP_LIB ${MKLML_IOMP_LIB})
|
|
|
|
|
SET(MKLDNN_IOMP_DIR ${MKLML_LIB_DIR})
|
|
|
|
|
ENDIF()
|
|
|
|
|
|
|
|
|
@ -59,7 +60,7 @@ ExternalProject_Add(
|
|
|
|
|
UPDATE_COMMAND ""
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
ADD_LIBRARY(mkldnn STATIC IMPORTED GLOBAL)
|
|
|
|
|
ADD_LIBRARY(mkldnn SHARED IMPORTED GLOBAL)
|
|
|
|
|
SET_PROPERTY(TARGET mkldnn PROPERTY IMPORTED_LOCATION ${MKLDNN_LIBRARY})
|
|
|
|
|
ADD_DEPENDENCIES(mkldnn ${MKLDNN_PROJECT})
|
|
|
|
|
MESSAGE(STATUS "Mkldnn library: ${MKLDNN_LIBRARY}")
|
|
|
|
|