File tree Expand file tree Collapse file tree 1 file changed +13
-2
lines changed
libs/qec/lib/decoders/plugins/trt_decoder Expand file tree Collapse file tree 1 file changed +13
-2
lines changed Original file line number Diff line number Diff 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+
118129if (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 )
133144endif ()
134145
You can’t perform that action at this time.
0 commit comments