Gitlab CPU CI: Use only CPU Jax

inducer · inducer · commit 029026cca465 · 2025-01-27T13:25:48.000-06:00
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:
