Skip to content

GitLab

Commit 587addb9 authored by Neelay Shah's avatar Neelay Shah Committed by GitHub
Browse files

build: adding framework as environment variable. Can use to select tests.

parent f1f29171
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
container/Dockerfile
View file @ 587addb9
...@@ -48,6 +48,7 @@ RUN pip install "git+https://github.com/triton-inference-server/perf_analyzer.gi ...@@ -48,6 +48,7 @@ RUN pip install "git+https://github.com/triton-inference-server/perf_analyzer.gi
# Backend & Framework Specific Installation # Backend & Framework Specific Installation
ARG FRAMEWORK="STANDARD" ARG FRAMEWORK="STANDARD"
ARG TENSORRTLLM_BACKEND_COMMIT= ARG TENSORRTLLM_BACKEND_COMMIT=
ENV FRAMEWORK=${FRAMEWORK}
RUN --mount=type=bind,source=./container/deps/requirements.tensorrtllm.txt,target=/tmp/requirements.txt \ RUN --mount=type=bind,source=./container/deps/requirements.tensorrtllm.txt,target=/tmp/requirements.txt \
--mount=type=bind,source=./container/deps/clone_tensorrtllm.sh,target=/tmp/clone_tensorrtllm.sh \ --mount=type=bind,source=./container/deps/clone_tensorrtllm.sh,target=/tmp/clone_tensorrtllm.sh \
if [[ "$FRAMEWORK" == "TENSORRTLLM" ]] ; then pip install --timeout=2000 -r /tmp/requirements.txt; /tmp/clone_tensorrtllm.sh --tensorrtllm-backend-commit ${TENSORRTLLM_BACKEND_COMMIT} ; fi if [[ "$FRAMEWORK" == "TENSORRTLLM" ]] ; then pip install --timeout=2000 -r /tmp/requirements.txt; /tmp/clone_tensorrtllm.sh --tensorrtllm-backend-commit ${TENSORRTLLM_BACKEND_COMMIT} ; fi
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment