Skip to content

Use ORT nightly in CI #287

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

Merged
justinchuby merged 16 commits into from
May 9, 2026
Merged

Use ORT nightly in CI #287

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
184cf24
Add additional dependencies to requirements.txt
justinchuby May 7, 2026
c4e9e32
Update CI requirements by removing unused packages
justinchuby May 7, 2026
7c8b98d
Update dependency installation in CI workflow
justinchuby May 7, 2026
02e065e
Refactor GPU E2E workflow by removing pip cache
justinchuby May 7, 2026
8826f64
Refactor GPU workflow by removing pip cache
justinchuby May 7, 2026
23e889a
Remove pip package caching from GPU validation workflow
justinchuby May 7, 2026
8e3786a
Update CI workflow dependencies and onnxruntime-gpu
justinchuby May 7, 2026
12f13ac
Apply suggestions from code review
justinchuby May 8, 2026
099f6bc
Fix ORT nightly install order and switch to --extra-index-url
Copilot May 8, 2026
5b4f752
Update gpu_l4_golden_parity.yml
justinchuby May 9, 2026
cb78474
Update PyTorch installation to CUDA 1.30
justinchuby May 9, 2026
ae5176d
Update main.yml
justinchuby May 9, 2026
8b8572b
Update validation_examples_gpu.yml
justinchuby May 9, 2026
5406b10
Apply suggestion from @justinchuby
justinchuby May 9, 2026
ba986ba
Apply suggestion from @justinchuby
justinchuby May 9, 2026
53fb00a
Merge branch 'main' into justinchuby-patch-1
justinchuby May 9, 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
13 changes: 3 additions & 10 deletions .github/workflows/gpu_l4_golden_parity.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,14 +43,6 @@ jobs:
with:
python-version: "3.12"

- name: Cache pip packages
uses: actions/cache@v5
with:
path: ~/.cache/pip
key: pip-gpu-${{ hashFiles('pyproject.toml', 'requirements/ci/requirements.txt') }}
restore-keys: |
pip-gpu-

- name: Cache HuggingFace models
uses: actions/cache@v5
with:
Expand All @@ -60,13 +52,14 @@ jobs:
hf-gpu-

- name: Install PyTorch (CUDA)
run: pip install torch --index-url https://download.pytorch.org/whl/cu124
run: pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130

- name: Install dependencies
run: |
pip install -r requirements/ci/requirements.txt
pip install onnxruntime-gpu
pip install soundfile librosa ml_dtypes flatbuffers numpy packaging protobuf sympy coloredlogs
pip install -e '.[testing,transformers]'
pip install --pre --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-13-nightly/pypi/simple/ onnxruntime-gpu onnxruntime-genai-cuda

- name: Check for golden test data
id: check_golden
Expand Down
13 changes: 3 additions & 10 deletions .github/workflows/gpu_l5_generation_e2e.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,14 +43,6 @@ jobs:
with:
python-version: "3.12"

- name: Cache pip packages
uses: actions/cache@v5
with:
path: ~/.cache/pip
key: pip-gpu-${{ hashFiles('pyproject.toml', 'requirements/ci/requirements.txt') }}
restore-keys: |
pip-gpu-

- name: Cache HuggingFace models
uses: actions/cache@v5
with:
Expand All @@ -60,13 +52,14 @@ jobs:
hf-gpu-

- name: Install PyTorch (CUDA)
run: pip install torch --index-url https://download.pytorch.org/whl/cu124
run: pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130

- name: Install dependencies
run: |
pip install -r requirements/ci/requirements.txt
pip install onnxruntime-gpu
pip install soundfile librosa ml_dtypes flatbuffers numpy packaging protobuf sympy coloredlogs
pip install -e '.[testing,transformers]'
pip install --pre --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-13-nightly/pypi/simple/ onnxruntime-gpu onnxruntime-genai-cuda

- name: Run L5 generation E2E tests
env:
Expand Down
5 changes: 3 additions & 2 deletions .github/workflows/main.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -277,12 +277,13 @@ jobs:
restore-keys: |
hf-integration-fast-
- name: Install PyTorch CUDA
run: pip install torch --index-url https://download.pytorch.org/whl/cu124
run: pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130
- name: Install dependencies
run: |
pip install -r requirements/ci/requirements.txt
pip install onnxruntime-gpu
pip install soundfile librosa ml_dtypes flatbuffers numpy packaging protobuf sympy coloredlogs
pip install -e '.[testing]'
pip install --pre --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-13-nightly/pypi/simple/ onnxruntime-gpu onnxruntime-genai-cuda
- name: Run fast integration tests
env:
HF_TOKEN: ${{ secrets.HF_TOKEN }}
Expand Down
16 changes: 4 additions & 12 deletions .github/workflows/validation_examples_gpu.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,14 +67,6 @@ jobs:
with:
python-version: "3.12"

- name: Cache pip packages
uses: actions/cache@v5
with:
path: ~/.cache/pip
key: pip-examples-gpu-${{ hashFiles('pyproject.toml', 'requirements/ci/requirements.txt') }}
restore-keys: |
pip-examples-gpu-

- name: Cache HuggingFace models
uses: actions/cache@v5
with:
Expand All @@ -84,14 +76,14 @@ jobs:
hf-examples-gpu-

- name: Install PyTorch (CUDA)
run: pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
run: pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130

- name: Install dependencies
run: |
pip install -r requirements/ci/requirements.txt
pip install -e '.[testing,transformers,ort-genai]'
pip install onnxruntime-gpu
pip install soundfile librosa ml_dtypes
pip install -e '.[testing,transformers]'
pip install soundfile librosa ml_dtypes flatbuffers numpy packaging protobuf sympy coloredlogs
Comment thread
justinchuby marked this conversation as resolved.
pip install --pre --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-13-nightly/pypi/simple/ onnxruntime-gpu onnxruntime-genai-cuda

- name: Run ${{ matrix.name }}
env:
Expand Down
Loading