fix

apache · Dec 7, 2019 · 91a8e3b · 91a8e3b
1 parent 7a42cd4
commit 91a8e3b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -177,9 +177,11 @@ else(MSVC)
     add_definitions(-DDMLC_USE_CXX11=1)
     add_definitions(-DDMLC_USE_CXX14=1)
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++14")
+    set(CMAKE_CUDA_STANDARD 14)
   elseif(SUPPORT_CXX11)
     add_definitions(-DDMLC_USE_CXX11=1)
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
+    set(CMAKE_CUDA_STANDARD 11)
   elseif(SUPPORT_CXX0X)
     add_definitions(-DDMLC_USE_CXX11=1)
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++0x")