Auto-detect CUDA arch in CMake GPU check (#666)

Compute capability 60 support is dropped from CUDA 13

Co-authored-by: Binyang Li <binyli@microsoft.com>
This commit is contained in:
Changho Hwang
2025-10-27 11:25:24 -07:00
committed by GitHub
parent 2b05908635
commit fc0aaaf1b4

View File

@@ -9,7 +9,7 @@ if(NOT CUDAToolkit_FOUND)
return()
endif()
set(CMAKE_CUDA_ARCHITECTURES "60")
set(CMAKE_CUDA_ARCHITECTURES native)
if(NOT CMAKE_CUDA_COMPILER)
# In case the CUDA Toolkit directory is not in the PATH
find_program(CUDA_COMPILER