fix: force $ORIGIN rpath for shared library files

abetlen · abetlen · commit d5f6a15a9b39 · 2024-07-01T23:03:26.000-04:00
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -28,6 +28,7 @@ if (LLAMA_BUILD)
     # which point to directories outside the build tree to the install RPATH
     set(CMAKE_INSTALL_RPATH_USE_LINK_PATH TRUE)
     set(CMAKE_INSTALL_RPATH "$ORIGIN")
+    set(CMAKE_SKIP_RPATH FALSE)
 
     # Building llama
     if (APPLE AND NOT CMAKE_SYSTEM_PROCESSOR MATCHES "arm64")
@@ -52,6 +53,10 @@ if (LLAMA_BUILD)
         FRAMEWORK DESTINATION ${LLAMA_CPP_PYTHON_INSTALL_DIR}
         RESOURCE DESTINATION ${LLAMA_CPP_PYTHON_INSTALL_DIR}
     )
+    set_target_properties(llama PROPERTIES
+        INSTALL_RPATH "$ORIGIN"
+        BUILD_WITH_INSTALL_RPATH TRUE
+    )
     install(
         TARGETS ggml
         LIBRARY DESTINATION ${LLAMA_CPP_PYTHON_INSTALL_DIR}
@@ -60,6 +65,10 @@ if (LLAMA_BUILD)
         FRAMEWORK DESTINATION ${LLAMA_CPP_PYTHON_INSTALL_DIR}
         RESOURCE DESTINATION ${LLAMA_CPP_PYTHON_INSTALL_DIR}
     )
+    set_target_properties(ggml PROPERTIES
+        INSTALL_RPATH "$ORIGIN"
+        BUILD_WITH_INSTALL_RPATH TRUE
+    )
     # Workaround for Windows + CUDA https://github.com/abetlen/llama-cpp-python/issues/563
     if (WIN32 AND (LLAMA_CUDA OR LLAMA_CUBLAS))
         install(
@@ -93,5 +102,9 @@ if (LLAMA_BUILD)
             FRAMEWORK DESTINATION ${LLAMA_CPP_PYTHON_INSTALL_DIR}
             RESOURCE DESTINATION ${LLAMA_CPP_PYTHON_INSTALL_DIR}
         )
+        set_target_properties(llava PROPERTIES
+            INSTALL_RPATH "$ORIGIN"
+            BUILD_WITH_INSTALL_RPATH TRUE
+        )
     endif()
 endif()
