Add numba-cuda-mlir conda package to cudf

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

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

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

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

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

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