pytorch

1
set(MKLDNN_USE_NATIVE_ARCH ${USE_NATIVE_ARCH})
2

3
if(CPU_AARCH64)
4
  include(${CMAKE_CURRENT_LIST_DIR}/ComputeLibrary.cmake)
5
endif()
6

7
find_package(MKLDNN QUIET)
8

9
if(NOT TARGET caffe2::mkldnn)
10
  add_library(caffe2::mkldnn INTERFACE IMPORTED)
11
endif()
12

13
set_property(
14
  TARGET caffe2::mkldnn PROPERTY INTERFACE_INCLUDE_DIRECTORIES
15
  ${MKLDNN_INCLUDE_DIR})
16
set_property(
17
  TARGET caffe2::mkldnn PROPERTY INTERFACE_LINK_LIBRARIES
18
  ${MKLDNN_LIBRARIES})
19