/usr/local/lib64/python3.6/site-packages/torch/share/cmake/Caffe2/public
NameSizeModeActions
cuda.cmake199010644editdlrm
gflags.cmake26200644editdlrm
glog.cmake23200644editdlrm
mkl.cmake2920644editdlrm
mkldnn.cmake4440644editdlrm
protobuf.cmake40030644editdlrm
threads.cmake5850644editdlrm
utils.cmake193380644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/share/cmake/Caffe2/public/mkldnn.cmake (444B)
set(MKLDNN_USE_NATIVE_ARCH ${USE_NATIVE_ARCH}) if(CPU_AARCH64) include(${CMAKE_CURRENT_LIST_DIR}/ComputeLibrary.cmake) endif() find_package(MKLDNN QUIET) if(NOT TARGET caffe2::mkldnn) add_library(caffe2::mkldnn INTERFACE IMPORTED) endif() set_property( TARGET caffe2::mkldnn PROPERTY INTERFACE_INCLUDE_DIRECTORIES ${MKLDNN_INCLUDE_DIR}) set_property( TARGET caffe2::mkldnn PROPERTY INTERFACE_LINK_LIBRARIES ${MKLDNN_LIBRARIES})