Skip to content

Commit 11a021d

Browse files
committed
Update the RPATH (again) for CUDA 13 changes
Signed-off-by: Ben Howe <[email protected]>
1 parent 7c0e0d3 commit 11a021d

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

libs/qec/lib/decoders/plugins/trt_decoder/CMakeLists.txt

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,10 +115,21 @@ set_target_properties(${MODULE_NAME} PROPERTIES
115115
# RPATH configuration
116116
# ==============================================================================
117117

118+
# Get the CUDA version (12 or 13)
119+
string(REGEX REPLACE "^([0-9]+)\\..*" "\\1" CUDA_MAJOR_VERSION ${CUDAToolkit_VERSION})
120+
message(STATUS "TRT CUDA version: ${CUDA_MAJOR_VERSION}")
121+
122+
string(CONCAT TRT_INSTALL_RPATH
123+
"$ORIGIN"
124+
":$ORIGIN/../../nvidia/cu${CUDA_MAJOR_VERSION}/lib"
125+
":$ORIGIN/../../lib"
126+
":$ORIGIN/../../../tensorrt_libs")
127+
message(STATUS "TRT_INSTALL_RPATH: ${TRT_INSTALL_RPATH}")
128+
118129
if (NOT SKBUILD)
119130
set_target_properties(${MODULE_NAME} PROPERTIES
120131
BUILD_RPATH "$ORIGIN:${TENSORRT_ROOT}/lib"
121-
INSTALL_RPATH "$ORIGIN:$ORIGIN/../../lib:$ORIGIN/../../../tensorrt_libs"
132+
INSTALL_RPATH "${TRT_INSTALL_RPATH}"
122133
)
123134

124135
# Let CMake automatically add paths of linked libraries to the RPATH:
@@ -128,7 +139,7 @@ else()
128139
# CUDA-Q install its libraries in site-packages/lib (or dist-packages/lib)
129140
# Thus, we need the $ORIGIN/../lib
130141
set_target_properties(${MODULE_NAME} PROPERTIES
131-
INSTALL_RPATH "$ORIGIN:$ORIGIN/../../lib:$ORIGIN/../../../tensorrt_libs"
142+
INSTALL_RPATH "${TRT_INSTALL_RPATH}"
132143
)
133144
endif()
134145

0 commit comments

Comments
 (0)