wip

joelberkeley · Jan 3, 2025 · 612cda3 · 612cda3
1 parent 90ab0bd
commit 612cda3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/checks.yml b/.github/workflows/checks.yml
@@ -300,7 +300,7 @@ jobs:
         run: |
           tar xfz tests-xla-cuda.tar.gz . && rm tests-xla-cuda.tar.gz
           docker load --input /tmp/xla-cuda.tar
-          docker run -v $(pwd):/xla-cuda -w /xla-cuda --gpus all xla-cuda sh -c "exit 0"  # needs a CUDA runner
+          docker run -v $(pwd):/xla-cuda -w /xla-cuda xla-cuda sh -c "exit 0"  # needs a CUDA runner
   readme:
     runs-on: ubuntu-24.04
     steps: