1*da0073e9SAndroid Build Coastguard Workerset(MKLDNN_USE_NATIVE_ARCH ${USE_NATIVE_ARCH}) 2*da0073e9SAndroid Build Coastguard Worker 3*da0073e9SAndroid Build Coastguard Workerif(CPU_AARCH64) 4*da0073e9SAndroid Build Coastguard Worker include(${CMAKE_CURRENT_LIST_DIR}/ComputeLibrary.cmake) 5*da0073e9SAndroid Build Coastguard Workerendif() 6*da0073e9SAndroid Build Coastguard Worker 7*da0073e9SAndroid Build Coastguard Workerfind_package(MKLDNN QUIET) 8*da0073e9SAndroid Build Coastguard Worker 9*da0073e9SAndroid Build Coastguard Workerif(NOT TARGET caffe2::mkldnn) 10*da0073e9SAndroid Build Coastguard Worker add_library(caffe2::mkldnn INTERFACE IMPORTED) 11*da0073e9SAndroid Build Coastguard Workerendif() 12*da0073e9SAndroid Build Coastguard Worker 13*da0073e9SAndroid Build Coastguard Workerset_property( 14*da0073e9SAndroid Build Coastguard Worker TARGET caffe2::mkldnn PROPERTY INTERFACE_INCLUDE_DIRECTORIES 15*da0073e9SAndroid Build Coastguard Worker ${MKLDNN_INCLUDE_DIR}) 16*da0073e9SAndroid Build Coastguard Workerset_property( 17*da0073e9SAndroid Build Coastguard Worker TARGET caffe2::mkldnn PROPERTY INTERFACE_LINK_LIBRARIES 18*da0073e9SAndroid Build Coastguard Worker ${MKLDNN_LIBRARIES}) 19