Skip to content

Commit 704a853

Browse files
committed
Fixes #694: Now using appropriate flags for all architectures when building fatbin files
1 parent c264b3f commit 704a853

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

examples/CMakeLists.txt

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@ if(NOT DEFINED CUDA_ARCH_FLAGS_STR)
1313
cmake_policy(SET CMP0104 OLD)
1414
endif()
1515
if((NOT DEFINED CUDA_ARCH_FLAGS) OR ("${CUDA_ARCH_FLAGS}" STREQUAL ""))
16-
cuda_select_nvcc_arch_flags(CUDA_ARCH_FLAGS_1 Auto)
17-
set(CUDA_ARCH_FLAGS ${CUDA_ARCH_FLAGS} CACHE STRING "CUDA -gencode parameters")
16+
cuda_select_nvcc_arch_flags(CUDA_ARCH_FLAGS_ Auto)
17+
set(CUDA_ARCH_FLAGS ${CUDA_ARCH_FLAGS_} CACHE STRING "CUDA -gencode parameters")
1818
string(REPLACE ";" " " CUDA_ARCH_FLAGS_STR_ "${CUDA_ARCH_FLAGS}")
1919
set(CUDA_ARCH_FLAGS_STR ${CUDA_ARCH_FLAGS_STR_} CACHE INTERNAL "CUDA Architecture flags")
2020
else()
@@ -121,7 +121,7 @@ IF((NOT ${CMAKE_SYSTEM_PROCESSOR} MATCHES "arm")
121121
OUTPUT memMapIPC_kernel.fatbin
122122
COMMAND ${CMAKE_CUDA_COMPILER} -fatbin
123123
${CCBIN_ARGUMENT}
124-
--generate-code arch=compute_${CMAKE_CUDA_ARCHITECTURES},code=sm_${CMAKE_CUDA_ARCHITECTURES}
124+
${CUDA_ARCH_FLAGS}
125125
-o ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/memMapIPC_kernel.fatbin
126126
${CMAKE_CURRENT_SOURCE_DIR}/modified_cuda_samples/memMapIPCDrv/memMapIPC_kernel.cu
127127
MAIN_DEPENDENCY modified_cuda_samples/memMapIPCDrv/memMapIPC_kernel.cu
@@ -133,7 +133,7 @@ IF((NOT ${CMAKE_SYSTEM_PROCESSOR} MATCHES "arm")
133133
OUTPUT vectorAddMMAP_kernel.fatbin
134134
COMMAND ${CMAKE_CUDA_COMPILER} -fatbin
135135
${CCBIN_ARGUMENT}
136-
--generate-code arch=compute_${CMAKE_CUDA_ARCHITECTURES},code=sm_${CMAKE_CUDA_ARCHITECTURES}
136+
${CUDA_ARCH_FLAGS}
137137
-o ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/vectorAddMMAP_kernel.fatbin
138138
${CMAKE_CURRENT_SOURCE_DIR}/modified_cuda_samples/vectorAddMMAP/vectorAdd_kernel.cu
139139
MAIN_DEPENDENCY modified_cuda_samples/vectorAddMMAP/vectorAdd_kernel.cu
@@ -147,7 +147,7 @@ add_custom_command(
147147
OUTPUT simpleTexture_kernel.fatbin
148148
COMMAND ${CMAKE_CUDA_COMPILER} -fatbin
149149
${CCBIN_ARGUMENT}
150-
--generate-code arch=compute_${CMAKE_CUDA_ARCHITECTURES},code=sm_${CMAKE_CUDA_ARCHITECTURES}
150+
${CUDA_ARCH_FLAGS}
151151
-o ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/simpleTexture_kernel.fatbin
152152
${CMAKE_CURRENT_SOURCE_DIR}/modified_cuda_samples/simpleTextureDrv/simpleTexture_kernel.cu
153153
MAIN_DEPENDENCY modified_cuda_samples/simpleTextureDrv/simpleTexture_kernel.cu

0 commit comments

Comments
 (0)