From 8a37bfa9d5da9e99831d602827832bf0f94b1f16 Mon Sep 17 00:00:00 2001 From: Kamil Tokarski Date: Fri, 12 Apr 2024 17:16:16 +0200 Subject: [PATCH] Bump up JAX version used in tests Signed-off-by: Kamil Tokarski --- qa/setup_packages.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/qa/setup_packages.py b/qa/setup_packages.py index 7d95208d80e..655f4edcd08 100755 --- a/qa/setup_packages.py +++ b/qa/setup_packages.py @@ -565,9 +565,10 @@ def get_pyvers_name(self, url, cuda_version): PckgVer( "0.4.13", python_min_ver="3.8", python_max_ver="3.8", dependencies=["jaxlib"] ), - # dax.fn.jax_function requires 0.4.16 which is the first one supporting + # dax.fn.jax_function requires at least 0.4.16 which is the first one supporting # `__dlpack__` method, while 0.4.13 is the last one supported with Python3.8 - PckgVer("0.4.16", python_min_ver="3.9", dependencies=["jaxlib"]), + # Use the most recent one avialable due to issues with address sanitizer run + PckgVer("0.4.26", python_min_ver="3.9", dependencies=["jaxlib"]), ] }, # name used during installation