Skip to content

Commit 024cbca

Browse files
committed
Don't use 'No CUDA devices found' as CUDA architecture
1 parent 414ba09 commit 024cbca

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

dependencies/FindO2GPU.cmake

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
# or submit itself to any jurisdiction.
1111

1212
# NOTE!!!! - Whenever this file is changed, move it over to alidist/resources
13-
# FindO2GPU.cmake Version 10
13+
# FindO2GPU.cmake Version 11
1414

1515
set(CUDA_COMPUTETARGET_DEFAULT_FULL 80-real 86-real 89-real 120-real 75-virtual)
1616
set(HIP_AMDGPUTARGET_DEFAULT_FULL gfx906;gfx908)
@@ -173,9 +173,7 @@ if(ENABLE_CUDA)
173173
message(${FAILURE_SEVERITY} "CUDA was found but cannot be enabled")
174174
set(CMAKE_CUDA_COMPILER OFF)
175175
endif()
176-
find_path(THRUST_INCLUDE_DIR thrust/version.h PATHS ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}
177-
PATH_SUFFIXES "" cccl
178-
NO_DEFAULT_PATH)
176+
find_path(THRUST_INCLUDE_DIR thrust/version.h PATHS ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES} PATH_SUFFIXES "" cccl NO_DEFAULT_PATH)
179177
if(THRUST_INCLUDE_DIR STREQUAL "THRUST_INCLUDE_DIR-NOTFOUND")
180178
message(${FAILURE_SEVERITY} "CUDA found but thrust not available, looked under: ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}")
181179
set(CMAKE_CUDA_COMPILER OFF)
@@ -188,7 +186,7 @@ if(ENABLE_CUDA)
188186
endif()
189187
endif()
190188
if(NOT CMAKE_CUDA_ARCHITECTURES OR O2_GPU_CUDA_UPDATE_NATIVE_ARCHITECTURE)
191-
if(NOT CMAKE_CUDA_ARCHITECTURES_NATIVE STREQUAL "")
189+
if(NOT CMAKE_CUDA_ARCHITECTURES_NATIVE STREQUAL "" AND NOT CMAKE_CUDA_ARCHITECTURES_NATIVE MATCHES "No CUDA devices found")
192190
set(CMAKE_CUDA_ARCHITECTURES ${CMAKE_CUDA_ARCHITECTURES_NATIVE})
193191
else()
194192
set(CMAKE_CUDA_ARCHITECTURES ${CUDA_COMPUTETARGET_DEFAULT_MINIMAL})

0 commit comments

Comments
 (0)