Skip to content

Commit 7e93f19

Browse files
committed
improve CUDA compiler detection in CMake
1 parent 1281d6f commit 7e93f19

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,7 @@ elseif(USE_ROCM)
116116
elseif(USE_CUDA)
117117
set(CMAKE_CUDA_STANDARD 17)
118118
find_package(CUDAToolkit REQUIRED)
119+
find_program(CMAKE_CUDA_COMPILER NAMES nvcc PATHS /usr/local/cuda/bin /opt/cuda/bin)
119120
add_compile_definitions("CUDA_MAJOR_VERSION=${CUDAToolkit_VERSION_MAJOR}")
120121

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

0 commit comments

Comments
 (0)