Skip to content

Commit 92e11db

Browse files
authored
fix(build): install pinned cuDNN Frontend Git ref in fw-final (#6447)
Signed-off-by: Ajay Balasa <abalasa@nvidia.com>
1 parent 253dca8 commit 92e11db

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

‎docker/Dockerfile.fw_final‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,11 +72,15 @@ RUN \
7272
BASE_RUNTIME_UV_ARGS="" && \
7373
if [ "${MBRIDGE_PRESERVE_BASE_RUNTIME:-False}" = "True" ]; then \
7474
BASE_RUNTIME_UV_ARGS="--no-install-package nvidia-cudnn-frontend --no-install-package transformer-engine --no-install-package transformer-engine-torch"; \
75+
elif [ -n "$CUDNN_FRONTEND_GIT_REF" ]; then \
76+
BASE_RUNTIME_UV_ARGS="--no-install-package nvidia-cudnn-frontend"; \
7577
fi && \
7678
uv sync --no-cache-dir --all-groups --inexact ${BASE_RUNTIME_UV_ARGS} && \
7779
if [ -n "$CUDNN_FRONTEND_GIT_REF" ] && [ "${MBRIDGE_PRESERVE_BASE_RUNTIME:-False}" != "True" ]; then \
78-
uv pip install --no-cache-dir --no-deps --reinstall \
79-
"nvidia-cudnn-frontend @ git+https://github.com/NVIDIA/cudnn-frontend.git@${CUDNN_FRONTEND_GIT_REF}"; \
80+
uv pip install --no-config --no-cache-dir --no-deps --reinstall \
81+
"nvidia-cudnn-frontend @ git+https://github.com/NVIDIA/cudnn-frontend.git@${CUDNN_FRONTEND_GIT_REF}" && \
82+
CUDNN_FRONTEND_GIT_REF="$CUDNN_FRONTEND_GIT_REF" python -c \
83+
'import importlib.metadata as m, json, os, sys; source = m.distribution("nvidia-cudnn-frontend").read_text("direct_url.json"); sys.exit(0 if json.loads(source)["vcs_info"]["commit_id"] == os.environ["CUDNN_FRONTEND_GIT_REF"] else "cuDNN Frontend Git ref mismatch")'; \
8084
fi && \
8185
python /opt/preserve_base_runtime.py check /opt/base-runtime-packages.json && \
8286
uv pip list && \

0 commit comments

Comments
 (0)