MegEngine/cmake/MKL_DNN.cmake

option(DNNL_BUILD_TESTS "" OFF)
option(DNNL_BUILD_EXAMPLES "" OFF)
set(DNNL_LIBRARY_TYPE STATIC CACHE STRING "config dnnl to STATIC")
# we do not want to use OMP now, so config to CPU mode
# if set to OMP, some dnnl algo will be more fast
set(DNNL_CPU_RUNTIME DNNL_RUNTIME_SEQ CACHE STRING "config dnnl to DNNL_RUNTIME_SEQ")
if(MGE_BLAS STREQUAL "MKL")
    option(_DNNL_USE_MKL "" ON)
    set(MKLROOT ${MKL_ROOT_DIR} CACHE STRING "MKL ROOT FOR DNNL")
    if (WIN32)
        set(MKLLIB "mkl_core" CACHE STRING "MKLLIB NAME FOR DNNL")
    else()
        set(MKLLIB "libmkl_core.a" CACHE STRING "MKLLIB NAME FOR DNNL")
    endif()
    # workround for DNNL link failed, we do not want to modify
    # dnnl MKL.cmake of intel-mkl-dnn when include intel-mkl-dnn
    # via add_subdirectory api
    link_directories(${MKL_ROOT_DIR}/lib)
else()
    option(_DNNL_USE_MKL "" OFF)
endif()

set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -Wno-unused-parameter -Wno-extra")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-parameter -Wno-extra")
add_subdirectory(${PROJECT_SOURCE_DIR}/third_party/intel-mkl-dnn)