enable cmake option WITH_MKLDNN

b5de21e8 · tensor-tang · 661826a7 · b5de21e8
显示空白变更内容
内联并排

Showing with 7 addition and 6 deletion

CMakeLists.txt CMakeLists.txt +7 -6

未找到文件。
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -117,13 +117,14 @@ else()
 endif()

 set(WITH_MKLML ${WITH_MKL})
-if (WITH_MKL AND AVX2_FOUND)
+if (NOT DEFINED WITH_MKLDNN)
+    if (WITH_MKL AND AVX2_FOUND)
        set(WITH_MKLDNN ON)
-else()
+    else()
        message(STATUS "Do not have AVX2 intrinsics and disabled MKL-DNN")
        set(WITH_MKLDNN OFF)
+    endif()
 endif()
-
 ########################################################################################

 include(external/mklml)     # download mklml package