Skip to content

ci: restore the image's cuDNN pin after reconciling requirements #1836

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
yueming-yuan wants to merge 3 commits into
base: main
Choose a base branch
from
+21 −1
Open

ci: restore the image's cuDNN pin after reconciling requirements #1836

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
bad0c0f
ci: restore the image's cuDNN pin after reconciling requirements
yueming-yuan Jul 28, 2026
47ecd5d
ci: do not let the empty-package probe trip errexit
yueming-yuan Jul 28, 2026
20e30ce
ci: guard the pip-show probe against pipefail
yueming-yuan Jul 28, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 21 additions & 1 deletion .github/workflows/_run-ci.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,27 @@ jobs:

- name: Reconcile Miles dependencies
shell: bash
run: python -m pip install -r "$GITHUB_WORKSPACE/requirements.txt" --break-system-packages
run: |
# The image pins cuDNN deliberately: transformer_engine's fused attention needs a
# newer build than torch's own exact pin, so the Dockerfile installs it last. This
# resolve drags it back down to whatever torch asks for, which makes fused attention
# backward fail with CUDNN_STATUS_BAD_PARAM. Record what the image shipped and put it
# back afterwards; --no-deps so nothing else in the resolved set moves.
pinned=""
for pkg in nvidia-cudnn-cu13 nvidia-cudnn-cu12; do
v=$(python -m pip show "$pkg" 2>/dev/null | awk '/^Version:/{print $2}' || true)
if [ -n "$v" ]; then pinned="$pinned $pkg==$v"; fi
done
python -m pip install -r "$GITHUB_WORKSPACE/requirements.txt" --break-system-packages
for spec in $pinned; do
pkg=${spec%%==*}
want=${spec##*==}
got=$(python -m pip show "$pkg" 2>/dev/null | awk '/^Version:/{print $2}' || true)
if [ -n "$got" ] && [ "$got" != "$want" ]; then
echo "Restoring image pin: $pkg $got -> $want"
python -m pip install --no-deps --break-system-packages "$spec"
fi
done

- name: Sync dependency sources
if: ${{ !inputs.skip_dependency_install }}
Expand Down
Loading