refactor: Update Onnxruntime library paths for aarch64 and other arch…

…itectures
locaal-ai · Sep 14, 2024 · 751ed97 · 751ed97
1 parent 3cb36c4
commit 751ed97
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/cmake/FetchOnnxruntime.cmake b/cmake/FetchOnnxruntime.cmake
@@ -84,11 +84,13 @@ elseif(MSVC)
 else()
   if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
     set(Onnxruntime_LINK_LIBS "${onnxruntime_SOURCE_DIR}/lib/libonnxruntime.so.${Onnxruntime_VERSION}")
-    set(Onnxruntime_INSTALL_LIBS ${Onnxruntime_LINK_LIBS})
+    set(Onnxruntime_INSTALL_LIBS ${Onnxruntime_LINK_LIBS} "${onnxruntime_SOURCE_DIR}/lib/libonnxruntime.so.1"
+                                 "${onnxruntime_SOURCE_DIR}/lib/libonnxruntime.so")
   else()
     set(Onnxruntime_LINK_LIBS "${onnxruntime_SOURCE_DIR}/lib/libonnxruntime.so.${Onnxruntime_VERSION}")
-    set(Onnxruntime_INSTALL_LIBS ${Onnxruntime_LINK_LIBS}
-                                 "${onnxruntime_SOURCE_DIR}/lib/libonnxruntime_providers_shared.so")
+    set(Onnxruntime_INSTALL_LIBS
+        ${Onnxruntime_LINK_LIBS} "${onnxruntime_SOURCE_DIR}/lib/libonnxruntime_providers_shared.so"
+        "${onnxruntime_SOURCE_DIR}/lib/libonnxruntime.so.1" "${onnxruntime_SOURCE_DIR}/lib/libonnxruntime.so")
   endif()
   target_link_libraries(${CMAKE_PROJECT_NAME} PRIVATE ${Onnxruntime_LINK_LIBS})
   target_include_directories(${CMAKE_PROJECT_NAME} SYSTEM PUBLIC "${onnxruntime_SOURCE_DIR}/include")