Commit c27397db authored by Andrey Morozov's avatar Andrey Morozov

minor fix

parent 789b02ad
...@@ -36,7 +36,9 @@ if (HAVE_CUDA) ...@@ -36,7 +36,9 @@ if (HAVE_CUDA)
include_directories(${CUDA_INCLUDE_DIRS}) include_directories(${CUDA_INCLUDE_DIRS})
link_directories(${CUDA_LIBRARIES}) link_directories(${CUDA_LIBRARIES})
set(CUDA_NVCC_FLAGS_ARCH "1.2 1.3 2.0" CACHE STRING "Add or remove compute copability" FORCE) set(CUDA_NVCC_FLAGS_ARCH " 1.0 1.1 1.2 1.3 2.0 ")
set(CUDA_NVCC_FLAGS_ARCHITECTURE ${CUDA_NVCC_FLAGS_ARCH} CACHE STRING "Add or remove compute copability" FORCE)
set(CUDA_NVCC_FLAGS_NUM "") set(CUDA_NVCC_FLAGS_NUM "")
...@@ -51,12 +53,9 @@ if (HAVE_CUDA) ...@@ -51,12 +53,9 @@ if (HAVE_CUDA)
set (loop_var "") set (loop_var "")
foreach( loop_var IN LISTS CUDA_NVCC_FLAGS_NUM) foreach( loop_var IN LISTS CUDA_NVCC_FLAGS_NUM)
set (CUDA_NVCC_FLAGS set (CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} -gencode arch=compute_${loop_var},code=sm_${loop_var})
"${CUDA_NVCC_FLAGS} -gencode=arch=compute_${loop_var},code=\"sm_${loop_var},compute_${loop_var}\"")
endforeach() endforeach()
set (CUDA_NVCC_FLAGS_ARCH ${CUDA_NVCC_FLAGS_ARCH_STR} CACHE STRING "Add or remove compute copability" FORCE)
message(STATUS " CUDA_NVCC_FLAGS: ${CUDA_NVCC_FLAGS} ") message(STATUS " CUDA_NVCC_FLAGS: ${CUDA_NVCC_FLAGS} ")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment