Added workaround for mshadow.cmake

apache · Aug 10, 2018 · 55ec400 · 55ec400
1 parent fe706b9
commit 55ec400
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -355,7 +355,13 @@ if(USE_CUDA AND FIRST_CUDA)
   include_directories(${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES})
 else()
   if(EXISTS ${CMAKE_CURRENT_SOURCE_DIR}/3rdparty/mshadow/cmake)
+    # workaroud to refrain mshadow from processing BLAS libraries
+    # BLAS libraries for MXNet are setup in cmake/ChooseBLAS.cmake
+    set(__BLAS ${BLAS})
+    set(BLAS "Override")
     include(3rdparty/mshadow/cmake/mshadow.cmake)
+    set(BLAS ${__BLAS})
+
     include(3rdparty/mshadow/cmake/Utils.cmake)
     include(3rdparty/mshadow/cmake/Cuda.cmake)
   else()