1set(MKLDNN_USE_NATIVE_ARCH ${USE_NATIVE_ARCH}) 2 3if(CPU_AARCH64) 4 include(${CMAKE_CURRENT_LIST_DIR}/ComputeLibrary.cmake) 5endif() 6 7find_package(MKLDNN QUIET) 8 9if(NOT TARGET caffe2::mkldnn) 10 add_library(caffe2::mkldnn INTERFACE IMPORTED) 11endif() 12 13set_property( 14 TARGET caffe2::mkldnn PROPERTY INTERFACE_INCLUDE_DIRECTORIES 15 ${MKLDNN_INCLUDE_DIR}) 16set_property( 17 TARGET caffe2::mkldnn PROPERTY INTERFACE_LINK_LIBRARIES 18 ${MKLDNN_LIBRARIES}) 19