Skip to content

Commit 1f6767c

Browse files
committed
GPU CMake: Another attempt to silence the CMake warnings about architectures
1 parent fef77bc commit 1f6767c

File tree

2 files changed

+8
-15
lines changed

2 files changed

+8
-15
lines changed

GPU/GPUTracking/Standalone/CMakeLists.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,10 +125,10 @@ find_package(O2GPU REQUIRED)
125125

126126
if(GPUCA_CONFIG_ONNX)
127127
find_package(onnxruntime REQUIRED)
128-
if(CUDA_ENABLED AND NOT DEFINED ORT_CUDA_BUILD)
129-
set(ORT_CUDA_BUILD ON)
130-
elseif(HIP_ENABLED AND NOT DEFINED ORT_ROCM_BUILD)
128+
if(HIP_ENABLED AND NOT DEFINED ORT_ROCM_BUILD)
131129
set(ORT_ROCM_BUILD ON)
130+
elseif(CUDA_ENABLED AND NOT DEFINED ORT_CUDA_BUILD)
131+
set(ORT_CUDA_BUILD ON)
132132
endif()
133133
else()
134134
set(onnxruntime_FOUND OFF)

dependencies/FindO2GPU.cmake

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,9 @@ endif()
104104
# ---------------------------------- CUDA ----------------------------------
105105
if(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 ----------------------------------
232229
if(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

Comments
 (0)