Skip to content

Commit 89cebea

Browse files
committed
GPU: Support virtual and real architectures in RTC
1 parent bfcff02 commit 89cebea

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

GPU/GPUTracking/Base/cuda/CMakeLists.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,11 @@ set(GPU_RTC_FLAGS "${CMAKE_CUDA_FLAGS} ${CMAKE_CUDA_FLAGS_${CMAKE_BUILD_TYPE_UPP
4141
set(GPU_RTC_FLAGS_ARCH "")
4242
if(CUDA_COMPUTETARGET)
4343
foreach(CUDA_ARCH ${CUDA_COMPUTETARGET})
44-
set(GPU_RTC_FLAGS_ARCH "${GPU_RTC_FLAGS_ARCH} -gencode arch=compute_${CUDA_ARCH},code=sm_${CUDA_ARCH}")
44+
string(REGEX REPLACE "-.*$" "" CUDA_ARCH_STRIPPED "${CUDA_ARCH}")
45+
set(GPU_RTC_FLAGS_ARCH "${GPU_RTC_FLAGS_ARCH} -gencode arch=compute_${CUDA_ARCH_STRIPPED},code=sm_${CUDA_ARCH_STRIPPED}")
4546
endforeach()
4647
list (GET CUDA_COMPUTETARGET 0 RTC_CUDA_ARCH)
48+
string(REGEX REPLACE "-.*$" "" RTC_CUDA_ARCH "${RTC_CUDA_ARCH}")
4749
set(RTC_CUDA_ARCH "${RTC_CUDA_ARCH}0")
4850
else()
4951
set(RTC_CUDA_ARCH "750")

0 commit comments

Comments
 (0)