run on tpus

google · Feb 26, 2024 · e9fbd71 · e9fbd71
1 parent 0e36803
commit e9fbd71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/BuildDockerImages.yml b/.github/workflows/BuildDockerImages.yml
@@ -28,7 +28,7 @@ on:
 
 jobs:
   build_images:
-    runs-on: ubuntu-latest
+    runs-on: ["self-hosted", "tpu", "${{ matrix.device-type }}"]
     steps:
     - uses: actions/checkout@v3
     - name: build jax stable image