Skip to content

Commit 978154c

Browse files
authored
CI: Move to self-hosted Windows GPU runners (#958)
Migrate the Windows testing to use the new NV GHA runners.
1 parent 0e72a17 commit 978154c

File tree

1 file changed

+5
-41
lines changed

1 file changed

+5
-41
lines changed

.github/workflows/test-wheel-windows.yml

Lines changed: 5 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -65,28 +65,23 @@ jobs:
6565
echo "MATRIX=${MATRIX}" | tee --append "${GITHUB_OUTPUT}"
6666
6767
test:
68-
# TODO: switch to this once the self-hosted runners are ready
69-
# name: py${{ matrix.PY_VER }}, ${{ matrix.CUDA_VER }}, ${{ (matrix.LOCAL_CTK == '1' && 'local') || 'wheels' }}, GPU ${{ matrix.GPU }}
70-
name: py${{ matrix.PY_VER }}, ${{ matrix.CUDA_VER }}, ${{ (matrix.LOCAL_CTK == '1' && 'local') || 'wheels' }}
68+
name: py${{ matrix.PY_VER }}, ${{ matrix.CUDA_VER }}, ${{ (matrix.LOCAL_CTK == '1' && 'local') || 'wheels' }}, GPU ${{ matrix.GPU }}
7169
# The build stage could fail but we want the CI to keep moving.
7270
needs: compute-matrix
7371
strategy:
7472
fail-fast: false
7573
matrix: ${{ fromJSON(needs.compute-matrix.outputs.MATRIX) }}
7674
if: ${{ github.repository_owner == 'nvidia' && !cancelled() }}
77-
# TODO: switch to self-hosted runners once they are ready
78-
# runs-on: "windows-${{ matrix.ARCH }}-gpu-${{ matrix.GPU }}-${{ matrix.DRIVER }}-1"
79-
runs-on: 'cuda-python-windows-gpu-github'
75+
runs-on: "windows-${{ matrix.ARCH }}-gpu-${{ matrix.GPU }}-${{ matrix.DRIVER }}-1"
8076
steps:
8177
- name: Checkout ${{ github.event.repository.name }}
8278
uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8 # v5.0.0
8379
with:
8480
fetch-depth: 0
8581

86-
# TODO: use setup-proxy-cache once we have self-hosted Windows runners
87-
# - name: Setup proxy cache
88-
# uses: nv-gha-runners/setup-proxy-cache@main
89-
# continue-on-error: true
82+
- name: Setup proxy cache
83+
uses: nv-gha-runners/setup-proxy-cache@main
84+
continue-on-error: true
9085

9186
- name: Update driver
9287
run: |
@@ -95,21 +90,6 @@ jobs:
9590
- name: Ensure GPU is working
9691
run: nvidia-smi
9792

98-
# TODO: remove this block once self-hosted runners are ready
99-
- name: Install Git for Windows
100-
# the GPU runner image does not have Git Bash pre-installed...
101-
env:
102-
# doesn't seem there's an easy way to avoid hard-coding it?
103-
GFW_EXE_URL: https://github.com/git-for-windows/git/releases/download/v2.49.0.windows.1/PortableGit-2.49.0-64-bit.7z.exe
104-
run: |
105-
Invoke-WebRequest -Uri "$env:GFW_EXE_URL" -OutFile "PortableGit.7z.exe"
106-
# Self-extracting, see https://gitforwindows.org/zip-archives-extracting-the-released-archives.html
107-
Start-Process .\PortableGit.7z.exe -Wait -Verbose -ArgumentList '-y -gm2'
108-
ls -l PortableGit
109-
echo "$((Get-Location).Path)\\PortableGit\\bin" >> $env:GITHUB_PATH
110-
$env:Path += ";$((Get-Location).Path)\\PortableGit\\bin"
111-
bash --version
112-
11393
- name: Set environment variables
11494
env:
11595
BUILD_CUDA_VER: ${{ inputs.build-ctk-ver }}
@@ -141,22 +121,6 @@ jobs:
141121
name: ${{ env.CUDA_BINDINGS_ARTIFACT_NAME }}
142122
path: ${{ env.CUDA_BINDINGS_ARTIFACTS_DIR }}
143123

144-
# TODO: remove this block once self-hosted runners are ready
145-
- name: Install gh cli
146-
# the GPU runner image does not have gh pre-installed...
147-
env:
148-
# doesn't seem there's an easy way to avoid hard-coding it?
149-
GH_MSI_URL: https://github.com/cli/cli/releases/download/v2.67.0/gh_2.67.0_windows_amd64.msi
150-
run: |
151-
Invoke-WebRequest -Uri "$env:GH_MSI_URL" -OutFile "gh_installer.msi"
152-
Start-Process msiexec.exe -Wait -Verbose -ArgumentList '/i "gh_installer.msi" /qn'
153-
$GH_POSSIBLE_PATHS = "C:\\Program Files\\GitHub CLI", "C:\\Program Files (x86)\\GitHub CLI"
154-
foreach ($p in $GH_POSSIBLE_PATHS) {
155-
echo "$p" >> $env:GITHUB_PATH
156-
$env:Path += ";$p"
157-
}
158-
gh --version
159-
160124
- name: Install zstd
161125
# the GPU runner image does not have zstd pre-installed... and it's needed by actions/cache
162126
if: ${{ matrix.LOCAL_CTK == '1' }}

0 commit comments

Comments
 (0)