Index: pytorch/cmake/public/mkldnn.cmake
===================================================================
--- pytorch.orig/cmake/public/mkldnn.cmake
+++ pytorch/cmake/public/mkldnn.cmake
@@ -5,6 +5,8 @@ if(CPU_AARCH64)
 endif()
 
 find_package(MKLDNN QUIET)
+find_library(MKLDNN_LIBRARIES dnnl)
+set(MKLDNN_FOUND TRUE)
 
 if(NOT TARGET caffe2::mkldnn)
   add_library(caffe2::mkldnn INTERFACE IMPORTED)
Index: pytorch/cmake/Modules/FindMKLDNN.cmake
===================================================================
--- pytorch.orig/cmake/Modules/FindMKLDNN.cmake
+++ pytorch/cmake/Modules/FindMKLDNN.cmake
@@ -16,13 +16,10 @@ IF(NOT MKLDNN_FOUND)
   SET(MKLDNN_LIBRARIES)
   SET(MKLDNN_INCLUDE_DIR)
 
-  SET(IDEEP_ROOT "${PROJECT_SOURCE_DIR}/third_party/ideep")
-  SET(MKLDNN_ROOT "${PROJECT_SOURCE_DIR}/third_party/ideep/mkl-dnn/third_party/oneDNN")
-  IF(NOT APPLE AND NOT WIN32 AND NOT BUILD_LITE_INTERPRETER)
-    MESSAGE("-- Will build oneDNN Graph")
-    SET(LLGA_ROOT "${PROJECT_SOURCE_DIR}/third_party/ideep/mkl-dnn")
-    SET(BUILD_ONEDNN_GRAPH ON)
-  ENDIF(NOT APPLE AND NOT WIN32 AND NOT BUILD_LITE_INTERPRETER)
+  set(IDEEP_ROOT "/usr")
+  set(MKLDNN_ROOT "/usr")
+  set(IDEEP_INCLUDE_DIR "/usr/include")
+  set(MKLDNN_INCLUDE_DIR "/usr/include")
 
   FIND_PACKAGE(BLAS)
   FIND_PATH(IDEEP_INCLUDE_DIR ideep.hpp PATHS ${IDEEP_ROOT} PATH_SUFFIXES include)
@@ -104,7 +101,9 @@ IF(NOT MKLDNN_FOUND)
       TARGET_COMPILE_OPTIONS(dnnl_graph PRIVATE -Wno-error=strict-overflow)
     ENDIF(CMAKE_COMPILER_IS_GNUCC)
   ELSE(BUILD_ONEDNN_GRAPH)
-    ADD_SUBDIRECTORY(${MKLDNN_ROOT})
+    #ADD_SUBDIRECTORY(${MKLDNN_ROOT})
+    find_package(dnnl REQUIRED)
+    add_library(dnnl ALIAS DNNL::dnnl)  
   ENDIF(BUILD_ONEDNN_GRAPH)
 
   IF(NOT TARGET dnnl)
@@ -112,11 +111,6 @@ IF(NOT MKLDNN_FOUND)
     RETURN()
   ENDIF(NOT TARGET dnnl)
 
-  IF(NOT APPLE AND CMAKE_COMPILER_IS_GNUCC)
-    TARGET_COMPILE_OPTIONS(dnnl PRIVATE -Wno-maybe-uninitialized)
-    TARGET_COMPILE_OPTIONS(dnnl PRIVATE -Wno-strict-overflow)
-    TARGET_COMPILE_OPTIONS(dnnl PRIVATE -Wno-error=strict-overflow)
-  ENDIF(NOT APPLE AND CMAKE_COMPILER_IS_GNUCC)
   LIST(APPEND MKLDNN_LIBRARIES ${MKL_OPENMP_LIBRARY})
   IF(BUILD_ONEDNN_GRAPH)
     LIST(APPEND MKLDNN_LIBRARIES "$<TARGET_FILE:dnnl_graph>")