Skip to content
Snippets Groups Projects
Commit ede7c140 authored by Markus Rampp's avatar Markus Rampp
Browse files

add cmake support for V100 GPUs/cuda9/compute_70

parent a53fc419
No related branches found
No related tags found
No related merge requests found
Pipeline #
...@@ -82,6 +82,9 @@ if (CUDA_FOUND) ...@@ -82,6 +82,9 @@ if (CUDA_FOUND)
if (NOT (${CUDA_VERSION} LESS 8.0)) if (NOT (${CUDA_VERSION} LESS 8.0))
list(APPEND CUDA_NVCC_FLAGS "-gencode=arch=compute_60,code=[sm_60,sm_61]") list(APPEND CUDA_NVCC_FLAGS "-gencode=arch=compute_60,code=[sm_60,sm_61]")
endif() endif()
if (NOT (${CUDA_VERSION} LESS 9.0))
list(APPEND CUDA_NVCC_FLAGS "-gencode=arch=compute_70,code=sm_70")
endif()
add_definitions(-DWITH_CUDA) add_definitions(-DWITH_CUDA)
set (BIOEM_CUDA_STATUS "Found") set (BIOEM_CUDA_STATUS "Found")
endif() endif()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment