Skip to content

Commit 44cc308

Browse files
dbarijoyang-nv
authored andcommitted
[https://nvbugs/5474037][fix] Fix building tritonbuild/tritonrelease images (#7157)
Signed-off-by: Dimitrios Bariamis <[email protected]> Signed-off-by: Dimitrios Bariamis <[email protected]> Co-authored-by: Dimitrios Bariamis <[email protected]> Signed-off-by: Wangshanshan <[email protected]>
1 parent ed4087a commit 44cc308

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

docker/Dockerfile.multi

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -174,9 +174,8 @@ FROM wheel AS tritonbuild
174174
WORKDIR /src/tensorrt_llm
175175
RUN pip install /src/tensorrt_llm/build/tensorrt_llm*.whl
176176
COPY ./triton_backend/ ./triton_backend/
177-
COPY ./jenkins/scripts/get_triton_tag.sh ./jenkins/scripts/get_triton_tag.sh
178-
COPY ./docker/Dockerfile.multi ./docker/Dockerfile.multi
179-
RUN bash ./triton_backend/inflight_batcher_llm/scripts/build.sh
177+
ARG TRITON_BASE_TAG
178+
RUN bash ./triton_backend/inflight_batcher_llm/scripts/build.sh -s "r${TRITON_BASE_TAG%-py3}"
180179

181180

182181
FROM release AS tritonrelease

0 commit comments

Comments
 (0)