fix cmake

apache · Mar 8, 2019 · a9b3886 · a9b3886
1 parent 849841d
commit a9b3886
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()