File tree Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Original file line number Diff line number Diff line change @@ -38,9 +38,13 @@ project(tritonmmdeploybackend LANGUAGES C CXX)
3838# doesn't use GPUs.
3939#
4040
41- set (TRITON_COMMON_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/common repo" )
42- set (TRITON_CORE_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/core repo" )
43- set (TRITON_BACKEND_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/backend repo" )
41+ if (NOT TRITON_TAG)
42+ set (TRITON_TAG main)
43+ endif ()
44+
45+ set (TRITON_COMMON_REPO_TAG ${TRITON_TAG} CACHE STRING "Tag for triton-inference-server/common repo" )
46+ set (TRITON_CORE_REPO_TAG ${TRITON_TAG} CACHE STRING "Tag for triton-inference-server/core repo" )
47+ set (TRITON_BACKEND_REPO_TAG ${TRITON_TAG} CACHE STRING "Tag for triton-inference-server/backend repo" )
4448
4549
4650#
@@ -73,6 +77,7 @@ add_library(triton-mmdeploy-backend SHARED
7377 model_state.cpp
7478 instance_state.cpp
7579 convert.cpp
80+ json_input.cpp
7681 mmdeploy.cpp)
7782
7883target_include_directories (triton-mmdeploy-backend PRIVATE ${CMAKE_CURRENT_SOURCE_DIR} )
@@ -98,7 +103,7 @@ target_link_libraries(triton-mmdeploy-backend PRIVATE MMDeployLibs)
98103
99104set_target_properties (triton-mmdeploy-backend PROPERTIES INSTALL_RPATH "\$ ORIGIN" )
100105
101- install (TARGETS triton-mmdeploy-backend DESTINATION backend /mmdeploy)
106+ install (TARGETS triton-mmdeploy-backend DESTINATION backends /mmdeploy)
102107
103108if (WIN32 )
104109 set_target_properties (
You can’t perform that action at this time.
0 commit comments