From 981a546703d129a6c4c521000bb07b2eb3fbf4c3 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 12 May 2022 19:42:27 -0500
Subject: [PATCH] Fix Jax installation on Gitlab CI
---
.gitlab-ci.yml | 10 +++++++---
1 file changed, 7 insertions(+), 3 deletions(-)
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 0436482..3f7e760 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
--
GitLab