Trying mcmodel=medium to fix the failure in CMake static build

AntiZpvoh · Apr 17, 2020 · 1af684c · 1af684c
1 parent 43f4b4e
commit 1af684c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -176,6 +176,9 @@ else()
     set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -O3")
   endif()
   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${CMAKE_C_FLAGS}")
+  if(USE_CUDA)
+    set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -mcmodel=medium")
+  endif()
 endif()
 
 if(NOT mxnet_LINKER_LIBS)