Skip to content

GitLab

Unverified Commit 37b3dbde authored by LJC00118's avatar LJC00118 Committed by GitHub
Browse files

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

* improve CUDA compiler detection in CMake

* Minor fix
parent 1281d6f8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
CMakeLists.txt
View file @ 37b3dbde
...@@ -116,6 +116,7 @@ elseif(USE_ROCM) ...@@ -116,6 +116,7 @@ elseif(USE_ROCM)
elseif(USE_CUDA) elseif(USE_CUDA)
set(CMAKE_CUDA_STANDARD 17) set(CMAKE_CUDA_STANDARD 17)
find_package(CUDAToolkit REQUIRED) find_package(CUDAToolkit REQUIRED)
set(CMAKE_CUDA_COMPILER "${CUDAToolkit_BIN_DIR}/nvcc")
add_compile_definitions("CUDA_MAJOR_VERSION=${CUDAToolkit_VERSION_MAJOR}") add_compile_definitions("CUDA_MAJOR_VERSION=${CUDAToolkit_VERSION_MAJOR}")
# Set `USE_CUDA=/usr/local/cuda-x.y` # Set `USE_CUDA=/usr/local/cuda-x.y`
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment