Gitlab CPU CI: Use only CPU Jax

illinois-ceesd · Jan 27, 2025 · 029026c · 029026c
1 parent 0903af2
commit 029026c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
@@ -2,6 +2,7 @@ Python 3 POCL:
   script: |
     export PYOPENCL_TEST=portable:cpu
     export EXTRA_INSTALL="jax[cpu]"
+    export JAX_PLATFORMS=cpu
     curl -L -O https://gitlab.tiker.net/inducer/ci-support/raw/main/build-and-test-py-project.sh
     . ./build-and-test-py-project.sh
   tags: