update the runner

ev-br · Feb 3, 2025 · d0ed5a6 · d0ed5a6
1 parent 421a662
commit d0ed5a6
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/.github/workflows/array-api-tests-cupy.yml b/.github/workflows/array-api-tests-cupy.yml
@@ -11,7 +11,7 @@ permissions:
 
 jobs:
   tests:
-    runs-on: ubuntu-latest
+    runs-on: ghcr.io/cirruslabs/ubuntu-runner-amd64-gpu:22.04
     strategy:
       matrix:
         python-version: ['3.11']
@@ -21,17 +21,25 @@ jobs:
       uses: actions/checkout@v4
       with:
         path: array-api-compat
+
     - name: Checkout array-api-tests
       uses: actions/checkout@v4
       with:
         repository: data-apis/array-api-tests
         submodules: 'true'
         path: array-api-tests
+
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/setup-python@v5
       with:
         python-version: ${{ matrix.python-version }}
 
+    - name: run nvidia-smi
+      run: nvidia-smi
+
+    - name: run nvidia-smi --query
+      run: nvidia-smi --query
+
     - name: Initialize submodules in -tests
       run: |
         cd array-api-tests