From 6ca1ee77332ab89ec60064ea4542fa68fed71b3c Mon Sep 17 00:00:00 2001 From: Pranav Sharma Date: Tue, 25 May 2021 17:36:15 -0700 Subject: [PATCH] Fix rpath issue with pybind. (#7829) * Fix rpath issue with pybind * Address PR comment --- cmake/onnxruntime.cmake | 3 +++ cmake/onnxruntime_python.cmake | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/cmake/onnxruntime.cmake b/cmake/onnxruntime.cmake index 256e07023a..b542490cee 100644 --- a/cmake/onnxruntime.cmake +++ b/cmake/onnxruntime.cmake @@ -75,6 +75,9 @@ elseif(onnxruntime_BUILD_APPLE_FRAMEWORK) ) else() onnxruntime_add_shared_library(onnxruntime ${CMAKE_CURRENT_BINARY_DIR}/generated_source.c) + if (onnxruntime_USE_CUDA) + set_property(TARGET onnxruntime APPEND_STRING PROPERTY LINK_FLAGS " -Xlinker -rpath=\\$ORIGIN") + endif() endif() add_dependencies(onnxruntime onnxruntime_generate_def ${onnxruntime_EXTERNAL_DEPENDENCIES}) diff --git a/cmake/onnxruntime_python.cmake b/cmake/onnxruntime_python.cmake index 419fdca22c..16103dca5d 100644 --- a/cmake/onnxruntime_python.cmake +++ b/cmake/onnxruntime_python.cmake @@ -143,7 +143,7 @@ elseif (APPLE) INSTALL_RPATH_USE_LINK_PATH FALSE) else() target_link_libraries(onnxruntime_pybind11_state PRIVATE ${onnxruntime_pybind11_state_libs} ${onnxruntime_EXTERNAL_LIBRARIES}) - set_property(TARGET onnxruntime_pybind11_state APPEND_STRING PROPERTY LINK_FLAGS " -Xlinker -rpath=\$ORIGIN") + set_property(TARGET onnxruntime_pybind11_state APPEND_STRING PROPERTY LINK_FLAGS " -Xlinker -rpath=\\$ORIGIN") endif() set_target_properties(onnxruntime_pybind11_state PROPERTIES PREFIX "")