|
|
|
@ -91,8 +91,13 @@ if(CUDA_FOUND) |
|
|
|
|
|
|
|
|
|
if(NOT DEFINED __cuda_arch_bin) |
|
|
|
|
if(ANDROID) |
|
|
|
|
set(__cuda_arch_bin "3.2") |
|
|
|
|
set(__cuda_arch_ptx "") |
|
|
|
|
if (ARM) |
|
|
|
|
set(__cuda_arch_bin "3.2") |
|
|
|
|
set(__cuda_arch_ptx "") |
|
|
|
|
elseif(AARCH64) |
|
|
|
|
set(__cuda_arch_bin "5.2") |
|
|
|
|
set(__cuda_arch_ptx "") |
|
|
|
|
endif() |
|
|
|
|
else() |
|
|
|
|
if(${CUDA_VERSION} VERSION_LESS "5.0") |
|
|
|
|
set(__cuda_arch_bin "1.1 1.2 1.3 2.0 2.1(2.0) 3.0") |
|
|
|
@ -154,7 +159,10 @@ if(CUDA_FOUND) |
|
|
|
|
|
|
|
|
|
if(ANDROID) |
|
|
|
|
set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} "-Xptxas;-dlcm=ca") |
|
|
|
|
set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} "-target-os-variant=Android") |
|
|
|
|
if(${CUDA_VERSION} VERSION_LESS "7.0") |
|
|
|
|
# since CUDA 7.0 OS variant is depricated |
|
|
|
|
set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} "-target-os-variant=Android") |
|
|
|
|
endif() |
|
|
|
|
endif() |
|
|
|
|
|
|
|
|
|
message(STATUS "CUDA NVCC target flags: ${CUDA_NVCC_FLAGS}") |
|
|
|
|