Bump up NUMBA version used in tests to 0.59.1

Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
NVIDIA · May 2, 2024 · 7ca79eb · 7ca79eb
1 parent 80b67f9
commit 7ca79eb
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git a/dali/test/python/test_utils.py b/dali/test/python/test_utils.py
@@ -947,9 +947,10 @@ def check_numba_compatibility_cpu(if_skip=True):
     #
     # TODO(michalz): Update the Numba version range when there's a fix - or possibly check
     # llvmlite directly (if still applicable)
-    if platform.processor().lower() in ("arm64", "aarch64", "armv8") and LooseVersion(
-        numba.__version__
-    ) >= LooseVersion("0.57.0"):
+    if platform.processor().lower() in ("arm64", "aarch64", "armv8") and (
+        LooseVersion(numba.__version__) >= LooseVersion("0.57.0")
+        and LooseVersion(numba.__version__) < LooseVersion("0.59.0")
+    ):
         if if_skip:
             raise SkipTest()
         else:

diff --git a/qa/setup_packages.py b/qa/setup_packages.py
@@ -575,7 +575,19 @@ def get_pyvers_name(self, url, cuda_version):
         links_index=("https://storage.googleapis.com/" "jax-releases/jax_cuda_releases.html"),
     ),
     CudaPackage(
-        "numba", {"110": [PckgVer("0.57.0", python_min_ver="3.8", dependencies=["numpy<1.24"])]}
+        "numba",
+        {
+            "110": [
+                # the more recent NUMBA doesn't support python 3.8 so keep it for this version here
+                PckgVer(
+                    "0.57.0",
+                    python_min_ver="3.8",
+                    python_max_ver="3.8",
+                    dependencies=["numpy<1.24"],
+                ),
+                PckgVer("0.59.1", python_min_ver="3.9", dependencies=["numpy<1.24"]),
+            ]
+        },
     ),
 ]