COMP: disable compute capabilities 2.0 for CUDA 9 (issue #112)

RTKConsortium · Oct 11, 2017 · 81b5fd7 · 81b5fd7
1 parent 7b0ea0e
commit 81b5fd7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cmake/FindCUDA_wrap.cmake b/cmake/FindCUDA_wrap.cmake
@@ -74,14 +74,20 @@ elseif("${CUDA_VERSION}" LESS 8.0)
      -gencode arch=compute_35,code=sm_35
      -gencode arch=compute_35,code=compute_35
      )
-else()
+elseif("${CUDA_VERSION}" LESS 9.0)
  set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS}
      -Wno-deprecated-gpu-targets
      -gencode arch=compute_20,code=sm_20
      -gencode arch=compute_30,code=sm_30
      -gencode arch=compute_35,code=sm_35
      -gencode arch=compute_35,code=compute_35
      )
+else()
+ set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS}
+     -gencode arch=compute_30,code=sm_30
+     -gencode arch=compute_35,code=sm_35
+     -gencode arch=compute_35,code=compute_35
+     )
 endif()
 
 if(CUDA_FOUND)