Skip to content

Commit f904435

Browse files
jbkyang-nviDavid Goodwin
authored and
David Goodwin
committed
add backend and core repo to cmake file (#3232)
1 parent 412dbe4 commit f904435

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

Dockerfile.sdk

+6
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@ ARG BASE_IMAGE=nvcr.io/nvidia/tritonserver:21.08-py3-min
2929

3030
ARG TRITON_CLIENT_REPO_SUBDIR=clientrepo
3131
ARG TRITON_COMMON_REPO_TAG=main
32+
ARG TRITON_CORE_REPO_TAG=main
33+
ARG TRITON_BACKEND_REPO_TAG=main
3234
ARG TRITON_THIRD_PARTY_REPO_TAG=main
3335
ARG TRITON_MODEL_ANALYZER_REPO_TAG=main
3436
ARG CMAKE_UBUNTU_VERSION=20.04
@@ -94,6 +96,8 @@ RUN rm -f /usr/bin/python && \
9496
# Build the client library and examples
9597
ARG TRITON_CLIENT_REPO_SUBDIR
9698
ARG TRITON_COMMON_REPO_TAG
99+
ARG TRITON_CORE_REPO_TAG
100+
ARG TRITON_BACKEND_REPO_TAG
97101
ARG TRITON_THIRD_PARTY_REPO_TAG
98102
ARG TRITON_ENABLE_GPU
99103

@@ -105,6 +109,8 @@ WORKDIR /workspace/build
105109
RUN cmake -DCMAKE_INSTALL_PREFIX=/workspace/install \
106110
-DTRITON_VERSION=`cat /workspace/TRITON_VERSION` \
107111
-DTRITON_COMMON_REPO_TAG=${TRITON_COMMON_REPO_TAG} \
112+
-DTRITON_CORE_REPO_TAG=${TRITON_CORE_REPO_TAG} \
113+
-DTRITON_BACKEND_REPO_TAG=${TRITON_BACKEND_REPO_TAG} \
108114
-DTRITON_THIRD_PARTY_REPO_TAG=${TRITON_THIRD_PARTY_REPO_TAG} \
109115
-DTRITON_ENABLE_CC_HTTP=ON -DTRITON_ENABLE_CC_GRPC=ON \
110116
-DTRITON_ENABLE_PYTHON_HTTP=ON -DTRITON_ENABLE_PYTHON_GRPC=ON \

0 commit comments

Comments
 (0)