diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml index 1a7cdc48ac7..9a40d91b0f2 100644 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -64,6 +64,7 @@ dependencies: - nbsphinx - ninja - notebook +- numba-cuda-mlir>=0.3.0 - numba-cuda>=0.22.2 - numba>=0.60.0,<0.65.0 - numexpr diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml index a25e0dc8163..14f219e9985 100644 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -64,6 +64,7 @@ dependencies: - nbsphinx - ninja - notebook +- numba-cuda-mlir>=0.3.0 - numba-cuda>=0.22.2 - numba>=0.60.0,<0.65.0 - numexpr diff --git a/conda/environments/all_cuda-133_arch-aarch64.yaml b/conda/environments/all_cuda-133_arch-aarch64.yaml index 875088c0549..4bac70b619d 100644 --- a/conda/environments/all_cuda-133_arch-aarch64.yaml +++ b/conda/environments/all_cuda-133_arch-aarch64.yaml @@ -64,6 +64,7 @@ dependencies: - nbsphinx - ninja - notebook +- numba-cuda-mlir>=0.3.0 - numba-cuda>=0.22.2 - numba>=0.60.0,<0.65.0 - numexpr diff --git a/conda/environments/all_cuda-133_arch-x86_64.yaml b/conda/environments/all_cuda-133_arch-x86_64.yaml index fcb76af8b6b..c141086bedb 100644 --- a/conda/environments/all_cuda-133_arch-x86_64.yaml +++ b/conda/environments/all_cuda-133_arch-x86_64.yaml @@ -64,6 +64,7 @@ dependencies: - nbsphinx - ninja - notebook +- numba-cuda-mlir>=0.3.0 - numba-cuda>=0.22.2 - numba>=0.60.0,<0.65.0 - numexpr diff --git a/conda/recipes/cudf/recipe.yaml b/conda/recipes/cudf/recipe.yaml index 29d60f9eedd..ad5e3e767ca 100644 --- a/conda/recipes/cudf/recipe.yaml +++ b/conda/recipes/cudf/recipe.yaml @@ -91,9 +91,7 @@ requirements: - pandas >=3.0.0,<3.1.0 - cupy >=14.0.1,!=14.1.0 - numba-cuda >=0.22.2 - # TODO: add `numba-cuda-mlir >=0.3.0` here once it is published to a conda - # channel. Currently only available via PyPI; the pip-only declaration - # lives in `dependencies.yaml::depends_on_numba_cuda_mlir`. + - numba-cuda-mlir >=0.3.0 - numba >=0.60.0,<0.65.0 - numpy >=2.0,<3.0 - pyarrow>=19.0.0,<24 # https://github.com/rapidsai/cudf/issues/22229 diff --git a/dependencies.yaml b/dependencies.yaml index 35044134c58..0cc75a93e86 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -1402,11 +1402,14 @@ dependencies: packages: - *numba_cuda depends_on_numba_cuda_mlir: - # numba-cuda-mlir is the MLIR-based numba-cuda compiler used by the cudf - # UDF backend. Currently only published to PyPI; conda packaging is a - # tracked feature request (see TODOs in conda/recipes/cudf/recipe.yaml). - # Once a conda package exists, add an `output_types: [conda]` entry here - # mirroring `depends_on_numba_cuda`. + common: + - output_types: [conda, requirements, pyproject] + packages: + # TODO: Remove after https://github.com/rapidsai/build-planning/issues/301 is resolved. + - numba>=0.60.0,<0.65.0 + - output_types: [conda] + packages: + - &numba_cuda_mlir numba-cuda-mlir>=0.3.0 specific: - output_types: [requirements, pyproject] matrices: @@ -1425,7 +1428,7 @@ dependencies: - matrix: cuda_suffixed: "false" packages: - - numba-cuda-mlir>=0.3.0 + - *numba_cuda_mlir depends_on_pylibcudf: common: - output_types: conda