Skip to content

Commit 146aa59

Browse files
authored
Merge pull request #2339 from ERGO-Code/find-cuda-win
FindCUDAConf on windows
2 parents cfa1614 + e9c99c3 commit 146aa59

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

cmake/FindCUDAConf.cmake

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,24 +2,37 @@ set(CUDA_LIBRARY-NOTFOUND, OFF)
22
message(NOTICE "Finding CUDA environment")
33
message(NOTICE " - CUDA Home detected at $ENV{CUDA_HOME}")
44
set(CMAKE_CUDA_ARCHITECTURES "all")
5-
set(CMAKE_CUDA_PATH "$ENV{CUDA_HOME}")
5+
6+
# On Windows users should set -DCMAKE_CUDA_PATH="..." when configuring CMake.
7+
# For all test setups CUPDLP_FIND_CUDA was not required on Windows.
8+
if (NOT WIN32)
9+
set(CMAKE_CUDA_PATH "$ENV{CUDA_HOME}")
10+
endif()
11+
# For local testing.
12+
# set(CMAKE_CUDA_PATH "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v12.9")
13+
14+
# message(STATUS "CMAKE_CUDA_PATH ${CMAKE_CUDA_PATH}")
15+
616
set(CMAKE_CUDA_COMPILER "${CMAKE_CUDA_PATH}/bin/nvcc")
717

818
enable_language(CUDA)
919

1020
find_library(CUDA_LIBRARY_ART
1121
NAMES cudart
1222
HINTS "${CMAKE_CUDA_PATH}/lib64/"
23+
HINTS "${CMAKE_CUDA_PATH}/lib/x64"
1324
REQUIRED
1425
)
1526
find_library(CUDA_LIBRARY_SPS
1627
NAMES cusparse
1728
HINTS "${CMAKE_CUDA_PATH}/lib64/"
29+
HINTS "${CMAKE_CUDA_PATH}/lib/x64/"
1830
REQUIRED
1931
)
2032
find_library(CUDA_LIBRARY_BLS
2133
NAMES cublas
2234
HINTS "${CMAKE_CUDA_PATH}/lib64/"
35+
HINTS "${CMAKE_CUDA_PATH}/lib/x64/"
2336
REQUIRED
2437
)
2538
if (${CUDA_LIBRARY-NOTFOUND})

0 commit comments

Comments
 (0)