[Enhancement] Improve CUDA compiler detection in CMake (#1054)

* improve CUDA compiler detection in CMake * Minor fix

[Enhancement] Improve CUDA compiler detection in CMake (#1054)
* improve CUDA compiler detection in CMake * Minor fix
37b3dbde · LJC00118 · GitHub · 1281d6f8 · 37b3dbde
Unverified Commit 37b3dbde authored Oct 17, 2025 by LJC00118 Committed by GitHub Oct 17, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

CMakeLists.txt CMakeLists.txt +1 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -116,6 +116,7 @@ elseif(USE_ROCM)
 elseif(USE_CUDA)
  set(CMAKE_CUDA_STANDARD 17)
  find_package(CUDAToolkit REQUIRED)
+  set(CMAKE_CUDA_COMPILER "${CUDAToolkit_BIN_DIR}/nvcc")
  add_compile_definitions("CUDA_MAJOR_VERSION=${CUDAToolkit_VERSION_MAJOR}")

  # Set `USE_CUDA=/usr/local/cuda-x.y`