From a9b3886d8e713cab777162e6a3e390387cbdcb18 Mon Sep 17 00:00:00 2001 From: taolv Date: Fri, 8 Mar 2019 10:09:56 +0800 Subject: [PATCH] fix cmake --- CMakeLists.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 6299c0f0d48e..7f3d326f1318 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -256,8 +256,8 @@ if(USE_MKLDNN) include_directories(3rdparty/mkldnn/include) include_directories(${PROJECT_BINARY_DIR}/3rdparty/mkldnn/include) - add_definitions(-DUSE_MKL=1) - add_definitions(-DCUB_MKL=1) + # add_definitions(-DUSE_MKL=1) + # add_definitions(-DCUB_MKL=1) add_definitions(-DMXNET_USE_MKLDNN=1) list(APPEND mxnet_LINKER_LIBS mkldnn) endif()