Skip to content

Commit 9fe6946

Browse files
committed
Fix linux build
1 parent 9d184df commit 9fe6946

File tree

1 file changed

+19
-12
lines changed

1 file changed

+19
-12
lines changed

CMakeLists.txt

Lines changed: 19 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -118,24 +118,24 @@ elseif(APPLE)
118118
)
119119
target_link_libraries(${CMAKE_PROJECT_NAME} PRIVATE onnxruntime::onnxruntime)
120120
elseif(MSVC)
121-
set(ONNXRUNTIME_ROOT "${CMAKE_SOURCE_DIR}/onnxruntime")
121+
set(ONNXRUNTIME_DIR "${CMAKE_SOURCE_DIR}/onnxruntime")
122122

123123
add_library(onnxruntime::onnxruntime_provider_shared SHARED IMPORTED)
124124
set_target_properties(
125125
onnxruntime::onnxruntime_provider_shared
126126
PROPERTIES
127-
IMPORTED_IMPLIB "${ONNXRUNTIME_ROOT}/lib/onnxruntime_providers_shared.lib"
128-
IMPORTED_LOCATION "${ONNXRUNTIME_ROOT}/lib/onnxruntime_providers_shared.dll"
129-
INTERFACE_INCLUDE_DIRECTORIES "${ONNXRUNTIME_ROOT}/include"
127+
IMPORTED_IMPLIB "${ONNXRUNTIME_DIR}/lib/onnxruntime_providers_shared.lib"
128+
IMPORTED_LOCATION "${ONNXRUNTIME_DIR}/lib/onnxruntime_providers_shared.dll"
129+
INTERFACE_INCLUDE_DIRECTORIES "${ONNXRUNTIME_DIR}/include"
130130
)
131131

132132
add_library(onnxruntime::onnxruntime SHARED IMPORTED)
133133
set_target_properties(
134134
onnxruntime::onnxruntime
135135
PROPERTIES
136-
IMPORTED_IMPLIB "${ONNXRUNTIME_ROOT}/lib/onnxruntime.lib"
137-
IMPORTED_LOCATION "${ONNXRUNTIME_ROOT}/lib/onnxruntime.dll"
138-
INTERFACE_INCLUDE_DIRECTORIES "${ONNXRUNTIME_ROOT}/include"
136+
IMPORTED_IMPLIB "${ONNXRUNTIME_DIR}/lib/onnxruntime.lib"
137+
IMPORTED_LOCATION "${ONNXRUNTIME_DIR}/lib/onnxruntime.dll"
138+
INTERFACE_INCLUDE_DIRECTORIES "${ONNXRUNTIME_DIR}/include"
139139
INTERFACE_LINK_OPTIONS "/DELAYLOAD:onnxruntime.dll"
140140
)
141141
target_link_libraries(onnxruntime::onnxruntime INTERFACE onnxruntime::onnxruntime_provider_shared)
@@ -145,16 +145,23 @@ elseif(MSVC)
145145

146146
install(
147147
FILES
148-
"${ONNXRUNTIME_ROOT}/lib/onnxruntime.dll"
149-
"${ONNXRUNTIME_ROOT}/lib/onnxruntime.pdb"
150-
"${ONNXRUNTIME_ROOT}/lib/onnxruntime_providers_shared.dll"
151-
"${ONNXRUNTIME_ROOT}/lib/onnxruntime_providers_shared.pdb"
148+
"${ONNXRUNTIME_DIR}/lib/onnxruntime.dll"
149+
"${ONNXRUNTIME_DIR}/lib/onnxruntime.pdb"
150+
"${ONNXRUNTIME_DIR}/lib/onnxruntime_providers_shared.dll"
151+
"${ONNXRUNTIME_DIR}/lib/onnxruntime_providers_shared.pdb"
152152
DESTINATION "${CMAKE_PROJECT_NAME}/bin/64bit/obs-backgroundremoval"
153153
)
154154
else()
155-
list(PREPEND CMAKE_PREFIX_PATH "${CMAKE_SOURCE_DIR}/onnxruntime")
155+
set(ONNXRUNTIME_DIR "${CMAKE_SOURCE_DIR}/onnxruntime")
156+
list(PREPEND CMAKE_PREFIX_PATH "${ONNXRUNTIME_DIR}")
156157
find_package(onnxruntime CONFIG REQUIRED)
157158
target_link_libraries(${CMAKE_PROJECT_NAME} PRIVATE onnxruntime::onnxruntime)
159+
file(GLOB ONNXRUNTIME_SO_FILES "${ONNXRUNTIME_DIR}/lib64/libonnxruntime.so.*")
160+
install(
161+
FILES ${ONNXRUNTIME_SO_FILES} "${ONNXRUNTIME_DIR}/lib64/libonnxruntime_providers_shared.so"
162+
DESTINATION ${CMAKE_INSTALL_LIBDIR}/obs-plugins/${CMAKE_PROJECT_NAME}
163+
)
164+
set_target_properties(${CMAKE_PROJECT_NAME} PROPERTIES INSTALL_RPATH "$ORIGIN/${CMAKE_PROJECT_NAME}")
158165
endif()
159166

160167
add_subdirectory(src/update-checker/CurlClient)

0 commit comments

Comments
 (0)