CI

pytorch · Jan 25, 2025 · d120cfe · d120cfe
1 parent 93db751
commit d120cfe
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/torch_xla2.yml b/.github/workflows/torch_xla2.yml
@@ -44,5 +44,6 @@ jobs:
         working-directory: torchax
         shell: bash
         run: |
+          export JAX_PLATFORMS=cpu
           pytest --forked test/
           XLA_FLAGS=--xla_force_host_platform_device_count=4 pytest -n 0 test_dist/