This commit is contained in:
iron 2026-01-03 00:19:50 +08:00 committed by GitHub
commit 1bdf60939d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions

View File

@ -52,6 +52,13 @@ if (CUDAToolkit_FOUND)
endif()
endif()
if(NOT DEFINED CMAKE_CUDA_COMPILER)
if(WIN32)
set(CMAKE_CUDA_COMPILER "${CUDAToolkit_BIN_DIR}/nvcc.exe")
else()
set(CMAKE_CUDA_COMPILER "${CUDAToolkit_BIN_DIR}/nvcc")
endif()
endif()
enable_language(CUDA)
# Replace any plain 12X CUDA architectures with their "architecture-specific" equivalents 12Xa.