Skip to content

Commit edbf29c

Browse files
committed
fix rpath issue
1 parent 42d4e8e commit edbf29c

1 file changed

Lines changed: 19 additions & 3 deletions

File tree

python/libcuopt/CMakeLists.txt

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,12 @@ project(
2525
LANGUAGES CXX CUDA
2626
)
2727

28+
# Debug: Print CUDA compiler version early
29+
message(STATUS "=============================================================")
30+
message(STATUS "libcuopt: CMAKE_CUDA_COMPILER_VERSION = ${CMAKE_CUDA_COMPILER_VERSION}")
31+
message(STATUS "libcuopt: CUDAToolkit_VERSION = ${CUDAToolkit_VERSION}")
32+
message(STATUS "=============================================================")
33+
2834
# Check if cuopt is already available. If so, it is the user's responsibility to ensure that the
2935
# CMake package is also available at build time of the Python cuopt package.
3036
find_package(cuopt "${CUOPT_VERSION}")
@@ -80,10 +86,20 @@ set(rpaths
8086
"$ORIGIN/../../nvidia/curand/lib"
8187
"$ORIGIN/../../nvidia/cusolver/lib"
8288
"$ORIGIN/../../nvidia/cusparse/lib"
83-
# CUDA version-specific paths
84-
$<$<VERSION_GREATER_EQUAL:${CMAKE_CUDA_COMPILER_VERSION},13.0>:"$ORIGIN/../../nvidia/cu13/lib">
85-
$<$<AND:$<VERSION_GREATER_EQUAL:${CMAKE_CUDA_COMPILER_VERSION},12.0>,$<VERSION_LESS:${CMAKE_CUDA_COMPILER_VERSION},13.0>>:"$ORIGIN/../../nvidia/cu12/lib">
8689
)
8790

91+
# Add CUDA version-specific paths based on CUDA compiler version
92+
message(STATUS "libcuopt: CMAKE_CUDA_COMPILER_VERSION = ${CMAKE_CUDA_COMPILER_VERSION}")
93+
if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 13.0 AND CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 14.0)
94+
message(STATUS "libcuopt: Adding cu13 RPATH")
95+
list(APPEND rpaths "$ORIGIN/../../nvidia/cu13/lib")
96+
elseif(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 12.0 AND CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 13.0)
97+
message(STATUS "libcuopt: Adding cu12 RPATH")
98+
list(APPEND rpaths "$ORIGIN/../../nvidia/cu12/lib")
99+
else()
100+
message(WARNING "libcuopt: Unsupported CUDA version ${CMAKE_CUDA_COMPILER_VERSION}")
101+
endif()
102+
message(STATUS "libcuopt: Final RPATH = ${rpaths}")
103+
88104
set_property(TARGET cuopt PROPERTY INSTALL_RPATH ${rpaths} APPEND)
89105
set_property(TARGET cuopt_cli PROPERTY INSTALL_RPATH ${rpaths} APPEND)

0 commit comments

Comments
 (0)