File tree Expand file tree Collapse file tree 1 file changed +14
-1
lines changed Expand file tree Collapse file tree 1 file changed +14
-1
lines changed Original file line number Diff line number Diff line change @@ -2,24 +2,37 @@ set(CUDA_LIBRARY-NOTFOUND, OFF)
2
2
message (NOTICE "Finding CUDA environment" )
3
3
message (NOTICE " - CUDA Home detected at $ENV{CUDA_HOME} " )
4
4
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
+
6
16
set (CMAKE_CUDA_COMPILER "${CMAKE_CUDA_PATH} /bin/nvcc" )
7
17
8
18
enable_language (CUDA )
9
19
10
20
find_library (CUDA_LIBRARY_ART
11
21
NAMES cudart
12
22
HINTS "${CMAKE_CUDA_PATH} /lib64/"
23
+ HINTS "${CMAKE_CUDA_PATH} /lib/x64"
13
24
REQUIRED
14
25
)
15
26
find_library (CUDA_LIBRARY_SPS
16
27
NAMES cusparse
17
28
HINTS "${CMAKE_CUDA_PATH} /lib64/"
29
+ HINTS "${CMAKE_CUDA_PATH} /lib/x64/"
18
30
REQUIRED
19
31
)
20
32
find_library (CUDA_LIBRARY_BLS
21
33
NAMES cublas
22
34
HINTS "${CMAKE_CUDA_PATH} /lib64/"
35
+ HINTS "${CMAKE_CUDA_PATH} /lib/x64/"
23
36
REQUIRED
24
37
)
25
38
if (${CUDA_LIBRARY-NOTFOUND} )
You can’t perform that action at this time.
0 commit comments