|
|
|
@ -16,10 +16,10 @@ file(GLOB MATH_HEADERS . *.h)
|
|
|
|
file(GLOB MATH_SOURCES . *.cpp)
|
|
|
|
file(GLOB MATH_SOURCES . *.cpp)
|
|
|
|
|
|
|
|
|
|
|
|
if(NOT WITH_MKLDNN)
|
|
|
|
if(NOT WITH_MKLDNN)
|
|
|
|
file(GLOB_RECURSE DNN_HEADER RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "MKLDNN*.h")
|
|
|
|
set(DNN_HEADER "${CMAKE_CURRENT_SOURCE_DIR}/MKLDNNMatrix.h")
|
|
|
|
file(GLOB_RECURSE DNN_SOURCES RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "MKLDNN*.cpp")
|
|
|
|
set(DNN_SOURCE "${CMAKE_CURRENT_SOURCE_DIR}/MKLDNNMatrix.cpp")
|
|
|
|
list(REMOVE_ITEM MATH_HEADERS ${DNN_HEADER})
|
|
|
|
list(REMOVE_ITEM MATH_HEADERS "${DNN_HEADER}")
|
|
|
|
list(REMOVE_ITEM MATH_SOURCES ${DNN_SOURCES})
|
|
|
|
list(REMOVE_ITEM MATH_SOURCES "${DNN_SOURCE}")
|
|
|
|
message(STATUS "Skip compiling with MKLDNNMatrix")
|
|
|
|
message(STATUS "Skip compiling with MKLDNNMatrix")
|
|
|
|
else()
|
|
|
|
else()
|
|
|
|
message(STATUS "Compile with MKLDNNMatrix")
|
|
|
|
message(STATUS "Compile with MKLDNNMatrix")
|
|
|
|
|