Fix test-util build to use correct BACKEND_REPO_TAG (#3063)

triton-inference-server · Jun 30, 2021 · c3de1d6 · c3de1d6
1 parent 30b08ba
commit c3de1d6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/build/CMakeLists.txt b/build/CMakeLists.txt
@@ -535,6 +535,7 @@ ExternalProject_Add(test-util
     -DCNMEM_PATH:PATH=${CMAKE_CURRENT_BINARY_DIR}/cnmem
     -DTRITON_COMMON_REPO_TAG:STRING=${TRITON_COMMON_REPO_TAG}
     -DTRITON_CORE_REPO_TAG:STRING=${TRITON_CORE_REPO_TAG}
+    -DTRITON_BACKEND_REPO_TAG:STRING=${TRITON_BACKEND_REPO_TAG}
     -DTRITON_ENABLE_GPU:BOOL=${TRITON_ENABLE_GPU}
     -DTRITON_MIN_COMPUTE_CAPABILITY:STRING=${TRITON_MIN_COMPUTE_CAPABILITY}
     -DTRITON_ENABLE_TENSORRT:BOOL=${TRITON_ENABLE_TENSORRT}

diff --git a/build/test-util/CMakeLists.txt b/build/test-util/CMakeLists.txt
@@ -57,9 +57,19 @@ FetchContent_Declare(
   GIT_TAG ${TRITON_CORE_REPO_TAG}
   GIT_SHALLOW ON
 )
+FetchContent_Declare(
+  repo-backend
+  GIT_REPOSITORY https://github.com/triton-inference-server/backend.git
+  GIT_TAG ${TRITON_BACKEND_REPO_TAG}
+  GIT_SHALLOW ON
+)
 
 set(TRITON_COMMON_ENABLE_PROTOBUF ON)
-FetchContent_MakeAvailable(repo-common repo-core)
+if(TRITON_ENABLE_GRPC)
+  set(TRITON_COMMON_ENABLE_GRPC ON)
+endif() # TRITON_ENABLE_GRPC
+
+FetchContent_MakeAvailable(repo-common repo-core repo-backend)
 
 if(${TRITON_ENABLE_GPU})
   add_definitions(-DTRITON_ENABLE_GPU=1)