diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index 0436482a9cf74caa159cc9c019df6a7b64039d80..3f7e7601d9f9995b2a2fd7044b99a0818771864a 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -16,10 +16,13 @@ Python 3 POCL: Python 3 Nvidia Titan V: script: | + curl -L -O https://tiker.net/ci-support-v0 + . ./ci-support-v0 export PYOPENCL_TEST=nvi:titan - export EXTRA_INSTALL="jax[cuda]" - curl -L -O https://gitlab.tiker.net/inducer/ci-support/raw/main/build-and-test-py-project.sh - . ./build-and-test-py-project.sh + build_py_project_in_venv + pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html + test_py_project + tags: - python3 - nvidia-titan-v @@ -82,6 +85,7 @@ Pylint: Mypy: script: | + EXTRA_INSTALL="jax[cpu]" curl -L -O https://tiker.net/ci-support-v0 . ./ci-support-v0 build_py_project_in_venv