Skip to content

Commit bfa3c64

Browse files
committed
update triton cmakelist
1 parent 0208fba commit bfa3c64

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

csrc/mmdeploy/triton/CMakeLists.txt

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff 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

7883
target_include_directories(triton-mmdeploy-backend PRIVATE ${CMAKE_CURRENT_SOURCE_DIR})
@@ -98,7 +103,7 @@ target_link_libraries(triton-mmdeploy-backend PRIVATE MMDeployLibs)
98103

99104
set_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

103108
if (WIN32)
104109
set_target_properties(

0 commit comments

Comments
 (0)