@@ -104,7 +104,9 @@ endif()
104104# ---------------------------------- CUDA ----------------------------------
105105if (ENABLE_CUDA)
106106 if (CUDA_COMPUTETARGET)
107- set (CMAKE_CUDA_ARCHITECTURES ${CUDA_COMPUTETARGET} CACHE STRING "" FORCE)
107+ set (CMAKE_CUDA_ARCHITECTURES ${CUDA_COMPUTETARGET} )
108+ else ()
109+ set (CMAKE_CUDA_ARCHITECTURES 61-virtual)
108110 endif ()
109111 set (CMAKE_CUDA_STANDARD ${CMAKE_CXX_STANDARD} )
110112 set (CMAKE_CUDA_STANDARD_REQUIRED TRUE )
@@ -121,11 +123,6 @@ if(ENABLE_CUDA)
121123 message (STATUS "Using as CUDA GCC version: ${GPUCA_CUDA_GCCBIN} " )
122124 set (CMAKE_CUDA_HOST_COMPILER "${GPUCA_CUDA_GCCBIN} " )
123125 endif ()
124- if (CUDA_COMPUTETARGET)
125- set (CMAKE_CUDA_ARCHITECTURES ${CUDA_COMPUTETARGET} CACHE STRING "" FORCE)
126- else ()
127- set (CMAKE_CUDA_ARCHITECTURES 61-virtual CACHE STRING "" FORCE)
128- endif ()
129126 enable_language (CUDA)
130127 get_property (LANGUAGES GLOBAL PROPERTY ENABLED_LANGUAGES )
131128 if (ENABLE_CUDA STREQUAL "AUTO" )
@@ -231,19 +228,15 @@ endif()
231228# ---------------------------------- HIP ----------------------------------
232229if (ENABLE_HIP)
233230 if (HIP_AMDGPUTARGET)
234- set (CMAKE_HIP_ARCHITECTURES "${HIP_AMDGPUTARGET} " CACHE STRING "" FORCE)
231+ set (CMAKE_HIP_ARCHITECTURES "${HIP_AMDGPUTARGET} " )
232+ set (AMDGPU_TARGETS "${HIP_AMDGPUTARGET} " )
235233 endif ()
236234 if (NOT "$ENV{CMAKE_PREFIX_PATH} " MATCHES "rocm" AND NOT CMAKE_PREFIX_PATH MATCHES "rocm" AND EXISTS "/opt/rocm/lib/cmake/" )
237235 list (APPEND CMAKE_PREFIX_PATH "/opt/rocm/lib/cmake" )
238236 endif ()
239237 if ("$ENV{CMAKE_PREFIX_PATH} " MATCHES "rocm" OR CMAKE_PREFIX_PATH MATCHES "rocm" )
240238 set (CMAKE_HIP_STANDARD ${CMAKE_CXX_STANDARD} )
241239 set (CMAKE_HIP_STANDARD_REQUIRED TRUE )
242- if (HIP_AMDGPUTARGET)
243- set (AMDGPU_TARGETS "${HIP_AMDGPUTARGET} " CACHE STRING "AMD GPU targets to compile for" FORCE)
244- set (GPU_TARGETS "${HIP_AMDGPUTARGET} " CACHE STRING "AMD GPU targets to compile for" FORCE)
245- set (CMAKE_HIP_ARCHITECTURES "${HIP_AMDGPUTARGET} " CACHE STRING "AMD GPU targets to compile for" FORCE)
246- endif ()
247240 set (TMP_ROCM_DIR_LIST "${CMAKE_PREFIX_PATH} :$ENV{CMAKE_PREFIX_PATH} " )
248241 string (REPLACE ":" ";" TMP_ROCM_DIR_LIST "${TMP_ROCM_DIR_LIST} " )
249242 list (FILTER TMP_ROCM_DIR_LIST INCLUDE REGEX rocm)
0 commit comments