Skip to content

Commit 4d2aa83

Browse files
committed
CI: run GPU driver install scripts from the workflow revision
Signed-off-by: Ralf Juengling <[email protected]>
1 parent 2e775e1 commit 4d2aa83

2 files changed

Lines changed: 18 additions & 2 deletions

File tree

‎.github/workflows/test-wheel-linux.yml‎

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,13 +146,21 @@ jobs:
146146
dependencies: "jq wget libgl1 libegl1 g++ util-linux yq"
147147
dependent_exes: "jq wget yq"
148148

149+
- name: Checkout driver install scripts (workflow revision)
150+
if: ${{ matrix.DRIVER != 'latest' && matrix.DRIVER != 'earliest' }}
151+
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
152+
with:
153+
ref: ${{ github.sha }}
154+
path: .driver-tools
155+
sparse-checkout: ci/tools
156+
149157
- name: Install GPU driver
150158
if: ${{ matrix.DRIVER != 'latest' && matrix.DRIVER != 'earliest' }}
151159
env:
152160
DRIVER: ${{ matrix.DRIVER }}
153161
DRIVER_SHA256: ${{ matrix.DRIVER_SHA256 }}
154162
GPU_TYPE: ${{ matrix.GPU }}
155-
run: ./ci/tools/install_gpu_driver.sh
163+
run: ./.driver-tools/ci/tools/install_gpu_driver.sh
156164

157165
- name: Ensure GPU is working
158166
run: nvidia-smi

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

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,14 +119,22 @@ jobs:
119119
with:
120120
enable-apt: true
121121

122+
- name: Checkout driver install scripts (workflow revision)
123+
if: ${{ matrix.DRIVER != 'latest' && matrix.DRIVER != 'earliest' }}
124+
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
125+
with:
126+
ref: ${{ github.sha }}
127+
path: .driver-tools
128+
sparse-checkout: ci/tools
129+
122130
- name: Install GPU driver
123131
if: ${{ matrix.DRIVER != 'latest' && matrix.DRIVER != 'earliest' }}
124132
env:
125133
DRIVER: ${{ matrix.DRIVER }}
126134
DRIVER_SHA256: ${{ matrix.DRIVER_SHA256 }}
127135
GPU_TYPE: ${{ matrix.GPU }}
128136
run: |
129-
ci/tools/install_gpu_driver.ps1
137+
.driver-tools/ci/tools/install_gpu_driver.ps1
130138
131139
- name: Configure driver mode
132140
env:

0 commit comments

Comments
 (0)