From b5de21e8d263c60826e05953eef37826e345feeb Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Wed, 16 May 2018 18:00:25 +0800 Subject: [PATCH] enable cmake option WITH_MKLDNN --- CMakeLists.txt | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 030bd19b3f..5ca9bc51cc 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -117,13 +117,14 @@ else() endif() set(WITH_MKLML ${WITH_MKL}) -if (WITH_MKL AND AVX2_FOUND) - set(WITH_MKLDNN ON) -else() - message(STATUS "Do not have AVX2 intrinsics and disabled MKL-DNN") - set(WITH_MKLDNN OFF) +if (NOT DEFINED WITH_MKLDNN) + if (WITH_MKL AND AVX2_FOUND) + set(WITH_MKLDNN ON) + else() + message(STATUS "Do not have AVX2 intrinsics and disabled MKL-DNN") + set(WITH_MKLDNN OFF) + endif() endif() - ######################################################################################## include(external/mklml) # download mklml package -- GitLab