diff --git a/conda/recipes/ucxx/recipe.yaml b/conda/recipes/ucxx/recipe.yaml index 2eb9287b..ee8386ab 100644 --- a/conda/recipes/ucxx/recipe.yaml +++ b/conda/recipes/ucxx/recipe.yaml @@ -260,6 +260,8 @@ outputs: - ${{ pin_subpackage("libucxx", exact=True) }} - cuda-cudart-dev run: + - numba >=0.59.1,<0.62.0a0 + - numba-cuda >=0.14.0,<0.15.0a0 - numpy >=1.23,<3.0a0 - pynvml >=12.0.0,<13.0.0a0 - python =${{ python }} @@ -269,7 +271,6 @@ outputs: - ${{ pin_subpackage("libucxx", exact=True) }} run_constraints: - cupy >=9.5.0 - - numba >=0.59.1,<0.62.0a0 ignore_run_exports: by_name: - cuda-cudart @@ -387,6 +388,8 @@ outputs: - setuptools - wheel run: + - numba >=0.59.1,<0.62.0a0 + - numba-cuda >=0.14.0,<0.15.0a0 - python - rapids-dask-dependency ${{ rapids_version }} - ${{ pin_subpackage("ucxx", upper_bound="x.x") }} diff --git a/dependencies.yaml b/dependencies.yaml index cf698618..d7d9cd53 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -285,6 +285,18 @@ dependencies: packages: - &numpy numpy>=1.23,<3.0a0 - pynvml>=12.0.0,<13.0.0a0 + - 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 run_python_distributed_ucxx: common: - output_types: [conda, requirements, pyproject] @@ -292,13 +304,13 @@ dependencies: - rapids-dask-dependency==25.10.*,>=0.0.0a0 - output_types: [conda] packages: - - &numba-cuda numba-cuda>=0.14.0,<0.15.0a0 + - *numba-cuda-dep specific: - output_types: [requirements, pyproject] matrices: - matrix: {cuda: "12.*"} packages: - - &numba-cuda-cu12-dep numba-cuda[cu12]>=0.14.0,<0.15.0a0 + - *numba-cuda-cu12-dep - matrix: # Fallback for no matrix packages: - *numba-cuda-cu12-dep @@ -316,18 +328,6 @@ dependencies: - pytest-asyncio>=1.0.0 - pytest-rerunfailures - rapids-dask-dependency==25.10.*,>=0.0.0a0 - - output_types: [conda] - packages: - - *numba-cuda - specific: - - output_types: [requirements, pyproject] - matrices: - - matrix: {cuda: "12.*"} - packages: - - *numba-cuda-cu12-dep - - matrix: # Fallback for no matrix - packages: - - *numba-cuda-cu12-dep test_python_distributed_ucxx: common: - output_types: [conda, requirements, pyproject] diff --git a/python/ucxx/pyproject.toml b/python/ucxx/pyproject.toml index 05e72834..8d7f7f00 100644 --- a/python/ucxx/pyproject.toml +++ b/python/ucxx/pyproject.toml @@ -20,6 +20,7 @@ license = { text = "BSD-3-Clause" } requires-python = ">=3.10" dependencies = [ "libucxx==0.46.*,>=0.0.0a0", + "numba-cuda[cu12]>=0.14.0,<0.15.0a0", "numpy>=1.23,<3.0a0", "pynvml>=12.0.0,<13.0.0a0", "rmm==25.10.*,>=0.0.0a0", @@ -41,7 +42,6 @@ test = [ "cloudpickle", "cudf==25.10.*,>=0.0.0a0", "cupy-cuda12x>=12.0.0", - "numba-cuda[cu12]>=0.14.0,<0.15.0a0", "pytest", "pytest-asyncio>=1.0.0", "pytest-rerunfailures",