Skip to content

CI - add Windows A100 runner #62

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

Draft
Alok-Joshi wants to merge 5 commits into
base: main
Choose a base branch
from
Draft

CI - add Windows A100 runner #62

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
69d5b61
CI - add Windows A100 runner to test matrix
Alok-Joshi Jul 28, 2026
0d8e69e
CI - split test job into test-linux and test-windows
Alok-Joshi Jul 29, 2026
9644f81
CI - install PyTorch on Windows test runner
Alok-Joshi Jul 29, 2026
bbd6b37
nit - use pip
Alok-Joshi Jul 29, 2026
74200ce
CI - add setup-python
Alok-Joshi Jul 29, 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
27 changes: 26 additions & 1 deletion .github/workflows/ci.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ jobs:
pip install hatch
hatch run docs:build

test:
test-linux:
name: test (${{ matrix.platform.runner }})
runs-on: ${{ matrix.platform.runner }}
strategy:
Expand All @@ -60,3 +60,28 @@ jobs:
run: |
pip install hatch
hatch run test_cu13:run

test-windows:
name: test (windows-amd64-gpu-a100-latest-1)
runs-on: windows-amd64-gpu-a100-latest-1
steps:
- uses: actions/checkout@v5
- uses: actions/setup-python@v6
with:
python-version: '3.12'
- name: "Install NCU 2026.2.1.0 (CUDA 13.3 Update 1)"
shell: pwsh
run: |
curl.exe -L -o ncu_installer.msi "https://developer.nvidia.com/downloads/assets/tools/secure/nsight-compute/2026_2_1/nsight_compute-windows-x86_64-2026.2.1.6.msi"
Start-Process msiexec.exe -Wait -ArgumentList '/i ncu_installer.msi /quiet /norestart'
Remove-Item ncu_installer.msi
"C:\Program Files\NVIDIA Corporation\Nsight Compute 2026.2.1" | Out-File -FilePath $env:GITHUB_PATH -Encoding utf8 -Append
- name: "Install PyTorch (CUDA 13.2)"
shell: pwsh
run: |
pip install torch --index-url https://download.pytorch.org/whl/cu132
- name: "Running Tests"
shell: pwsh
run: |
pip install hatch
hatch run test_cu13:run
Loading