diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml index 62b9c86d..b4d4e660 100644 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -14,6 +14,7 @@ dependencies: - distributed-ucxx==0.46.*,>=0.0.0a0 - kvikio==25.10.*,>=0.0.0a0 - numactl-devel-cos7-aarch64 +- numba-cuda>=0.14.0,<0.15.0a0 - numba>=0.59.1,<0.62.0a0 - numpy>=1.23,<3.0a0 - numpydoc>=1.1.0 diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml index 2b5152c0..863d7f20 100644 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -14,6 +14,7 @@ dependencies: - distributed-ucxx==0.46.*,>=0.0.0a0 - kvikio==25.10.*,>=0.0.0a0 - numactl-devel-cos7-x86_64 +- numba-cuda>=0.14.0,<0.15.0a0 - numba>=0.59.1,<0.62.0a0 - numpy>=1.23,<3.0a0 - numpydoc>=1.1.0 diff --git a/conda/recipes/dask-cuda/recipe.yaml b/conda/recipes/dask-cuda/recipe.yaml index 87e3e831..0df2d04c 100644 --- a/conda/recipes/dask-cuda/recipe.yaml +++ b/conda/recipes/dask-cuda/recipe.yaml @@ -37,6 +37,7 @@ requirements: - python - click >=8.1 - numba >=0.59.1,<0.62.0a0 + - numba-cuda >=0.14.0,<0.15.0a0 - numpy >=1.23,<3.0a0 - pandas >=1.3 - pynvml >=12.0.0,<13.0.0a0 diff --git a/dependencies.yaml b/dependencies.yaml index 6fb776eb..ed3dee64 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -148,6 +148,18 @@ dependencies: - pynvml>=12.0.0,<13.0.0a0 - rapids-dask-dependency==25.10.*,>=0.0.0a0 - zict>=2.0.0 + - output_types: [conda] + packages: + - &numba-cuda-dep numba-cuda>=0.14.0,<0.15.0a0 + specific: + - output_types: [requirements, pyproject] + matrices: + - matrix: {cuda: "12.*"} + packages: + - &numba-cuda-cu12-dep numba-cuda[cu12]>=0.14.0,<0.15.0a0 + - matrix: # Fallback for no matrix + packages: + - *numba-cuda-cu12-dep test_python: common: - output_types: [conda, requirements, pyproject] diff --git a/pyproject.toml b/pyproject.toml index a7ecb263..a8beccee 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -18,6 +18,7 @@ license = { text = "Apache-2.0" } requires-python = ">=3.10" dependencies = [ "click >=8.1", + "numba-cuda[cu12]>=0.14.0,<0.15.0a0", "numba>=0.59.1,<0.62.0a0", "numpy>=1.23,<3.0a0", "pandas>=1.3",