Skip to content

Commit ab78702

Browse files
authored
Make TRITON_BASE_TAG visible in tritonbuild
Signed-off-by: Dimitrios Bariamis <[email protected]>
1 parent 253af9f commit ab78702

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

docker/Dockerfile.multi

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,7 @@ FROM wheel AS tritonbuild
176176
WORKDIR /src/tensorrt_llm
177177
RUN pip install /src/tensorrt_llm/build/tensorrt_llm*.whl
178178
COPY ./triton_backend/ ./triton_backend/
179+
ARG TRITON_BASE_TAG
179180
RUN bash ./triton_backend/inflight_batcher_llm/scripts/build.sh -s "r${TRITON_BASE_TAG%-py3}"
180181

181182

0 commit comments

Comments
 (0)