Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Bump up NUMBA version used in tests to 0.59.1 #5451

Merged
merged 2 commits into from
May 9, 2024
Merged

Bump up NUMBA version used in tests to 0.59.1 #5451

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
7ca79eb
Bump up NUMBA version used in tests to 0.59.1
JanuszL May 1, 2024
5e4dae7
Review fix
JanuszL May 6, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions dali/test/python/test_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
stiepan marked this conversation as resolved.
Show resolved Hide resolved
# 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:
Expand Down
14 changes: 13 additions & 1 deletion qa/setup_packages.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"]),
]
},
),
]

Expand Down
Loading