From 309cf31d7c45f35131ba73459d99c0897546f23b Mon Sep 17 00:00:00 2001 From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Date: Fri, 14 Aug 2026 23:22:38 +0000 Subject: [PATCH 1/3] Add numba-cuda-mlir conda package to cudf --- .../all_cuda-129_arch-aarch64.yaml | 124 ------------------ .../all_cuda-129_arch-x86_64.yaml | 124 ------------------ .../all_cuda-133_arch-aarch64.yaml | 124 ------------------ .../all_cuda-133_arch-x86_64.yaml | 124 ------------------ conda/recipes/cudf/recipe.yaml | 4 +- dependencies.yaml | 15 ++- 6 files changed, 10 insertions(+), 505 deletions(-) delete mode 100644 conda/environments/all_cuda-129_arch-aarch64.yaml delete mode 100644 conda/environments/all_cuda-129_arch-x86_64.yaml delete mode 100644 conda/environments/all_cuda-133_arch-aarch64.yaml delete mode 100644 conda/environments/all_cuda-133_arch-x86_64.yaml diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml deleted file mode 100644 index 1a7cdc48ac7..00000000000 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ /dev/null @@ -1,124 +0,0 @@ -# This file is generated by `rapids-dependency-file-generator`. -# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. -channels: -- rapidsai-nightly -- rapidsai -- conda-forge -dependencies: -- aiobotocore>=2.2.0,<3 -- binutils -- boto3>=1.21.21 -- botocore>=1.24.21 -- breathe>=4.35.0 -- c-compiler -- cachetools -- clang-tools==20.1.8 -- clang==20.1.8 -- cmake>=4.0 -- cramjam -- croaring==4.4.2 -- cuda-bindings>=12.9.2,<13.0 -- cuda-cudart-dev -- cuda-cupti-dev -- cuda-nvcc -- cuda-nvrtc-dev -- cuda-nvtx-dev -- cuda-profiler-api -- cuda-sanitizer-api -- cuda-version=12.9 -- cupy>=14.0.1,!=14.1.0 -- cxx-compiler -- cython>=3.2.2,<3.3.0a0 -- dask-cuda==26.10.*,>=0.0.0a0 -- dlpack>=0.8,<1.0 -- doxygen=1.9.1 -- fastavro>=0.22.9 -- flatbuffers==24.3.25 -- fsspec>=0.6.0 -- gcc_linux-aarch64=14.* -- gdb -- hypothesis>=6.131.7 -- identify>=2.5.20 -- include-what-you-use==0.24.0 -- ipykernel -- ipython -- jupyter_client -- libcufile-dev -- libcurand-dev -- libkvikio==26.10.*,>=0.0.0a0 -- libnvcomp-dev==5.3.0.16 -- libnvjitlink-dev -- librapidsmpf==26.10.*,>=0.0.0a0 -- librdkafka<2.15.0a0 -- librmm==26.10.*,>=0.0.0a0 -- libucxx==0.52.*,>=0.0.0a0 -- make -- matplotlib-base<3.11 -- mmh3 -- moto>=4.0.8 -- msgpack-python -- myst-nb -- nanoarrow -- nbconvert -- nbformat -- nbsphinx -- ninja -- notebook -- numba-cuda>=0.22.2 -- numba>=0.60.0,<0.65.0 -- numexpr -- numpy>=2.0,<3.0 -- numpydoc -- nvidia-ml-py>=12 -- nvtx>=0.2.1 -- odfpy -- openpyxl -- packaging -- pandas>=3.0.0,<3.0.4a0 -- pandoc -- polars>=1.35,<1.43 -- pre-commit -- psutil -- pyarrow>=19.0.0,<24 -- pytables -- pytest-benchmark -- pytest-cases>=3.8.2 -- pytest-cov -- pytest-httpserver -- pytest-rerunfailures!=16.0.0 -- pytest-xdist -- pytest<9.1.0 -- python-calamine -- python-confluent-kafka -- python-xxhash -- python>=3.11 -- pytorch>=2.4.0 -- pyxlsb -- rapids-build-backend>=0.4.0,<0.5.0 -- rapids-dask-dependency==26.10.*,>=0.0.0a0 -- rapids-logger==0.3.* -- rapidsmpf==26.10.*,>=0.0.0a0 -- ray-default>=2.55.1 -- rich -- rmm==26.10.*,>=0.0.0a0 -- s3fs>=2022.3.0 -- scikit-build-core>=0.11.0 -- scipy -- sphinx-autobuild -- sphinx-copybutton -- sphinx-markdown-tables -- sphinx-remove-toctrees -- sphinx>=8.1.0 -- sphinxcontrib-websupport -- streamz -- structlog -- sysroot_linux-aarch64==2.28 -- typing_extensions>=4.0.0 -- ucxx==0.52.*,>=0.0.0a0 -- xlrd -- xlsxwriter -- zlib>=1.2.13 -- zstandard -- pip: - - nvidia-sphinx-theme -name: all_cuda-129_arch-aarch64 diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml deleted file mode 100644 index a25e0dc8163..00000000000 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ /dev/null @@ -1,124 +0,0 @@ -# This file is generated by `rapids-dependency-file-generator`. -# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. -channels: -- rapidsai-nightly -- rapidsai -- conda-forge -dependencies: -- aiobotocore>=2.2.0,<3 -- binutils -- boto3>=1.21.21 -- botocore>=1.24.21 -- breathe>=4.35.0 -- c-compiler -- cachetools -- clang-tools==20.1.8 -- clang==20.1.8 -- cmake>=4.0 -- cramjam -- croaring==4.4.2 -- cuda-bindings>=12.9.2,<13.0 -- cuda-cudart-dev -- cuda-cupti-dev -- cuda-nvcc -- cuda-nvrtc-dev -- cuda-nvtx-dev -- cuda-profiler-api -- cuda-sanitizer-api -- cuda-version=12.9 -- cupy>=14.0.1,!=14.1.0 -- cxx-compiler -- cython>=3.2.2,<3.3.0a0 -- dask-cuda==26.10.*,>=0.0.0a0 -- dlpack>=0.8,<1.0 -- doxygen=1.9.1 -- fastavro>=0.22.9 -- flatbuffers==24.3.25 -- fsspec>=0.6.0 -- gcc_linux-64=14.* -- gdb -- hypothesis>=6.131.7 -- identify>=2.5.20 -- include-what-you-use==0.24.0 -- ipykernel -- ipython -- jupyter_client -- libcufile-dev -- libcurand-dev -- libkvikio==26.10.*,>=0.0.0a0 -- libnvcomp-dev==5.3.0.16 -- libnvjitlink-dev -- librapidsmpf==26.10.*,>=0.0.0a0 -- librdkafka<2.15.0a0 -- librmm==26.10.*,>=0.0.0a0 -- libucxx==0.52.*,>=0.0.0a0 -- make -- matplotlib-base<3.11 -- mmh3 -- moto>=4.0.8 -- msgpack-python -- myst-nb -- nanoarrow -- nbconvert -- nbformat -- nbsphinx -- ninja -- notebook -- numba-cuda>=0.22.2 -- numba>=0.60.0,<0.65.0 -- numexpr -- numpy>=2.0,<3.0 -- numpydoc -- nvidia-ml-py>=12 -- nvtx>=0.2.1 -- odfpy -- openpyxl -- packaging -- pandas>=3.0.0,<3.0.4a0 -- pandoc -- polars>=1.35,<1.43 -- pre-commit -- psutil -- pyarrow>=19.0.0,<24 -- pytables -- pytest-benchmark -- pytest-cases>=3.8.2 -- pytest-cov -- pytest-httpserver -- pytest-rerunfailures!=16.0.0 -- pytest-xdist -- pytest<9.1.0 -- python-calamine -- python-confluent-kafka -- python-xxhash -- python>=3.11 -- pytorch>=2.4.0 -- pyxlsb -- rapids-build-backend>=0.4.0,<0.5.0 -- rapids-dask-dependency==26.10.*,>=0.0.0a0 -- rapids-logger==0.3.* -- rapidsmpf==26.10.*,>=0.0.0a0 -- ray-default>=2.55.1 -- rich -- rmm==26.10.*,>=0.0.0a0 -- s3fs>=2022.3.0 -- scikit-build-core>=0.11.0 -- scipy -- sphinx-autobuild -- sphinx-copybutton -- sphinx-markdown-tables -- sphinx-remove-toctrees -- sphinx>=8.1.0 -- sphinxcontrib-websupport -- streamz -- structlog -- sysroot_linux-64==2.28 -- typing_extensions>=4.0.0 -- ucxx==0.52.*,>=0.0.0a0 -- xlrd -- xlsxwriter -- zlib>=1.2.13 -- zstandard -- pip: - - nvidia-sphinx-theme -name: all_cuda-129_arch-x86_64 diff --git a/conda/environments/all_cuda-133_arch-aarch64.yaml b/conda/environments/all_cuda-133_arch-aarch64.yaml deleted file mode 100644 index 875088c0549..00000000000 --- a/conda/environments/all_cuda-133_arch-aarch64.yaml +++ /dev/null @@ -1,124 +0,0 @@ -# This file is generated by `rapids-dependency-file-generator`. -# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. -channels: -- rapidsai-nightly -- rapidsai -- conda-forge -dependencies: -- aiobotocore>=2.2.0,<3 -- binutils -- boto3>=1.21.21 -- botocore>=1.24.21 -- breathe>=4.35.0 -- c-compiler -- cachetools -- clang-tools==20.1.8 -- clang==20.1.8 -- cmake>=4.0 -- cramjam -- croaring==4.4.2 -- cuda-bindings>=13.0.1,<14.0 -- cuda-cudart-dev -- cuda-cupti-dev -- cuda-nvcc -- cuda-nvrtc-dev -- cuda-nvtx-dev -- cuda-profiler-api -- cuda-sanitizer-api -- cuda-version=13.3 -- cupy>=14.0.1,!=14.1.0 -- cxx-compiler -- cython>=3.2.2,<3.3.0a0 -- dask-cuda==26.10.*,>=0.0.0a0 -- dlpack>=0.8,<1.0 -- doxygen=1.9.1 -- fastavro>=0.22.9 -- flatbuffers==24.3.25 -- fsspec>=0.6.0 -- gcc_linux-aarch64=14.* -- gdb -- hypothesis>=6.131.7 -- identify>=2.5.20 -- include-what-you-use==0.24.0 -- ipykernel -- ipython -- jupyter_client -- libcufile-dev -- libcurand-dev -- libkvikio==26.10.*,>=0.0.0a0 -- libnvcomp-dev==5.3.0.16 -- libnvjitlink-dev -- librapidsmpf==26.10.*,>=0.0.0a0 -- librdkafka<2.15.0a0 -- librmm==26.10.*,>=0.0.0a0 -- libucxx==0.52.*,>=0.0.0a0 -- make -- matplotlib-base<3.11 -- mmh3 -- moto>=4.0.8 -- msgpack-python -- myst-nb -- nanoarrow -- nbconvert -- nbformat -- nbsphinx -- ninja -- notebook -- numba-cuda>=0.22.2 -- numba>=0.60.0,<0.65.0 -- numexpr -- numpy>=2.0,<3.0 -- numpydoc -- nvidia-ml-py>=12 -- nvtx>=0.2.1 -- odfpy -- openpyxl -- packaging -- pandas>=3.0.0,<3.0.4a0 -- pandoc -- polars>=1.35,<1.43 -- pre-commit -- psutil -- pyarrow>=19.0.0,<24 -- pytables -- pytest-benchmark -- pytest-cases>=3.8.2 -- pytest-cov -- pytest-httpserver -- pytest-rerunfailures!=16.0.0 -- pytest-xdist -- pytest<9.1.0 -- python-calamine -- python-confluent-kafka -- python-xxhash -- python>=3.11 -- pytorch>=2.10.0 -- pyxlsb -- rapids-build-backend>=0.4.0,<0.5.0 -- rapids-dask-dependency==26.10.*,>=0.0.0a0 -- rapids-logger==0.3.* -- rapidsmpf==26.10.*,>=0.0.0a0 -- ray-default>=2.55.1 -- rich -- rmm==26.10.*,>=0.0.0a0 -- s3fs>=2022.3.0 -- scikit-build-core>=0.11.0 -- scipy -- sphinx-autobuild -- sphinx-copybutton -- sphinx-markdown-tables -- sphinx-remove-toctrees -- sphinx>=8.1.0 -- sphinxcontrib-websupport -- streamz -- structlog -- sysroot_linux-aarch64==2.28 -- typing_extensions>=4.0.0 -- ucxx==0.52.*,>=0.0.0a0 -- xlrd -- xlsxwriter -- zlib>=1.2.13 -- zstandard -- pip: - - nvidia-sphinx-theme -name: all_cuda-133_arch-aarch64 diff --git a/conda/environments/all_cuda-133_arch-x86_64.yaml b/conda/environments/all_cuda-133_arch-x86_64.yaml deleted file mode 100644 index fcb76af8b6b..00000000000 --- a/conda/environments/all_cuda-133_arch-x86_64.yaml +++ /dev/null @@ -1,124 +0,0 @@ -# This file is generated by `rapids-dependency-file-generator`. -# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. -channels: -- rapidsai-nightly -- rapidsai -- conda-forge -dependencies: -- aiobotocore>=2.2.0,<3 -- binutils -- boto3>=1.21.21 -- botocore>=1.24.21 -- breathe>=4.35.0 -- c-compiler -- cachetools -- clang-tools==20.1.8 -- clang==20.1.8 -- cmake>=4.0 -- cramjam -- croaring==4.4.2 -- cuda-bindings>=13.0.1,<14.0 -- cuda-cudart-dev -- cuda-cupti-dev -- cuda-nvcc -- cuda-nvrtc-dev -- cuda-nvtx-dev -- cuda-profiler-api -- cuda-sanitizer-api -- cuda-version=13.3 -- cupy>=14.0.1,!=14.1.0 -- cxx-compiler -- cython>=3.2.2,<3.3.0a0 -- dask-cuda==26.10.*,>=0.0.0a0 -- dlpack>=0.8,<1.0 -- doxygen=1.9.1 -- fastavro>=0.22.9 -- flatbuffers==24.3.25 -- fsspec>=0.6.0 -- gcc_linux-64=14.* -- gdb -- hypothesis>=6.131.7 -- identify>=2.5.20 -- include-what-you-use==0.24.0 -- ipykernel -- ipython -- jupyter_client -- libcufile-dev -- libcurand-dev -- libkvikio==26.10.*,>=0.0.0a0 -- libnvcomp-dev==5.3.0.16 -- libnvjitlink-dev -- librapidsmpf==26.10.*,>=0.0.0a0 -- librdkafka<2.15.0a0 -- librmm==26.10.*,>=0.0.0a0 -- libucxx==0.52.*,>=0.0.0a0 -- make -- matplotlib-base<3.11 -- mmh3 -- moto>=4.0.8 -- msgpack-python -- myst-nb -- nanoarrow -- nbconvert -- nbformat -- nbsphinx -- ninja -- notebook -- numba-cuda>=0.22.2 -- numba>=0.60.0,<0.65.0 -- numexpr -- numpy>=2.0,<3.0 -- numpydoc -- nvidia-ml-py>=12 -- nvtx>=0.2.1 -- odfpy -- openpyxl -- packaging -- pandas>=3.0.0,<3.0.4a0 -- pandoc -- polars>=1.35,<1.43 -- pre-commit -- psutil -- pyarrow>=19.0.0,<24 -- pytables -- pytest-benchmark -- pytest-cases>=3.8.2 -- pytest-cov -- pytest-httpserver -- pytest-rerunfailures!=16.0.0 -- pytest-xdist -- pytest<9.1.0 -- python-calamine -- python-confluent-kafka -- python-xxhash -- python>=3.11 -- pytorch>=2.10.0 -- pyxlsb -- rapids-build-backend>=0.4.0,<0.5.0 -- rapids-dask-dependency==26.10.*,>=0.0.0a0 -- rapids-logger==0.3.* -- rapidsmpf==26.10.*,>=0.0.0a0 -- ray-default>=2.55.1 -- rich -- rmm==26.10.*,>=0.0.0a0 -- s3fs>=2022.3.0 -- scikit-build-core>=0.11.0 -- scipy -- sphinx-autobuild -- sphinx-copybutton -- sphinx-markdown-tables -- sphinx-remove-toctrees -- sphinx>=8.1.0 -- sphinxcontrib-websupport -- streamz -- structlog -- sysroot_linux-64==2.28 -- typing_extensions>=4.0.0 -- ucxx==0.52.*,>=0.0.0a0 -- xlrd -- xlsxwriter -- zlib>=1.2.13 -- zstandard -- pip: - - nvidia-sphinx-theme -name: all_cuda-133_arch-x86_64 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 From 8f3209a377f18eb7ac9a65226b079cd2efc2632f Mon Sep 17 00:00:00 2001 From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Date: Fri, 14 Aug 2026 23:23:31 +0000 Subject: [PATCH 2/3] Undo deletion of conda/environments --- .../all_cuda-129_arch-aarch64.yaml | 124 ++++++++++++++++++ .../all_cuda-129_arch-x86_64.yaml | 124 ++++++++++++++++++ .../all_cuda-133_arch-aarch64.yaml | 124 ++++++++++++++++++ .../all_cuda-133_arch-x86_64.yaml | 124 ++++++++++++++++++ 4 files changed, 496 insertions(+) create mode 100644 conda/environments/all_cuda-129_arch-aarch64.yaml create mode 100644 conda/environments/all_cuda-129_arch-x86_64.yaml create mode 100644 conda/environments/all_cuda-133_arch-aarch64.yaml create mode 100644 conda/environments/all_cuda-133_arch-x86_64.yaml diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml new file mode 100644 index 00000000000..1a7cdc48ac7 --- /dev/null +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -0,0 +1,124 @@ +# This file is generated by `rapids-dependency-file-generator`. +# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. +channels: +- rapidsai-nightly +- rapidsai +- conda-forge +dependencies: +- aiobotocore>=2.2.0,<3 +- binutils +- boto3>=1.21.21 +- botocore>=1.24.21 +- breathe>=4.35.0 +- c-compiler +- cachetools +- clang-tools==20.1.8 +- clang==20.1.8 +- cmake>=4.0 +- cramjam +- croaring==4.4.2 +- cuda-bindings>=12.9.2,<13.0 +- cuda-cudart-dev +- cuda-cupti-dev +- cuda-nvcc +- cuda-nvrtc-dev +- cuda-nvtx-dev +- cuda-profiler-api +- cuda-sanitizer-api +- cuda-version=12.9 +- cupy>=14.0.1,!=14.1.0 +- cxx-compiler +- cython>=3.2.2,<3.3.0a0 +- dask-cuda==26.10.*,>=0.0.0a0 +- dlpack>=0.8,<1.0 +- doxygen=1.9.1 +- fastavro>=0.22.9 +- flatbuffers==24.3.25 +- fsspec>=0.6.0 +- gcc_linux-aarch64=14.* +- gdb +- hypothesis>=6.131.7 +- identify>=2.5.20 +- include-what-you-use==0.24.0 +- ipykernel +- ipython +- jupyter_client +- libcufile-dev +- libcurand-dev +- libkvikio==26.10.*,>=0.0.0a0 +- libnvcomp-dev==5.3.0.16 +- libnvjitlink-dev +- librapidsmpf==26.10.*,>=0.0.0a0 +- librdkafka<2.15.0a0 +- librmm==26.10.*,>=0.0.0a0 +- libucxx==0.52.*,>=0.0.0a0 +- make +- matplotlib-base<3.11 +- mmh3 +- moto>=4.0.8 +- msgpack-python +- myst-nb +- nanoarrow +- nbconvert +- nbformat +- nbsphinx +- ninja +- notebook +- numba-cuda>=0.22.2 +- numba>=0.60.0,<0.65.0 +- numexpr +- numpy>=2.0,<3.0 +- numpydoc +- nvidia-ml-py>=12 +- nvtx>=0.2.1 +- odfpy +- openpyxl +- packaging +- pandas>=3.0.0,<3.0.4a0 +- pandoc +- polars>=1.35,<1.43 +- pre-commit +- psutil +- pyarrow>=19.0.0,<24 +- pytables +- pytest-benchmark +- pytest-cases>=3.8.2 +- pytest-cov +- pytest-httpserver +- pytest-rerunfailures!=16.0.0 +- pytest-xdist +- pytest<9.1.0 +- python-calamine +- python-confluent-kafka +- python-xxhash +- python>=3.11 +- pytorch>=2.4.0 +- pyxlsb +- rapids-build-backend>=0.4.0,<0.5.0 +- rapids-dask-dependency==26.10.*,>=0.0.0a0 +- rapids-logger==0.3.* +- rapidsmpf==26.10.*,>=0.0.0a0 +- ray-default>=2.55.1 +- rich +- rmm==26.10.*,>=0.0.0a0 +- s3fs>=2022.3.0 +- scikit-build-core>=0.11.0 +- scipy +- sphinx-autobuild +- sphinx-copybutton +- sphinx-markdown-tables +- sphinx-remove-toctrees +- sphinx>=8.1.0 +- sphinxcontrib-websupport +- streamz +- structlog +- sysroot_linux-aarch64==2.28 +- typing_extensions>=4.0.0 +- ucxx==0.52.*,>=0.0.0a0 +- xlrd +- xlsxwriter +- zlib>=1.2.13 +- zstandard +- pip: + - nvidia-sphinx-theme +name: all_cuda-129_arch-aarch64 diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml new file mode 100644 index 00000000000..a25e0dc8163 --- /dev/null +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -0,0 +1,124 @@ +# This file is generated by `rapids-dependency-file-generator`. +# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. +channels: +- rapidsai-nightly +- rapidsai +- conda-forge +dependencies: +- aiobotocore>=2.2.0,<3 +- binutils +- boto3>=1.21.21 +- botocore>=1.24.21 +- breathe>=4.35.0 +- c-compiler +- cachetools +- clang-tools==20.1.8 +- clang==20.1.8 +- cmake>=4.0 +- cramjam +- croaring==4.4.2 +- cuda-bindings>=12.9.2,<13.0 +- cuda-cudart-dev +- cuda-cupti-dev +- cuda-nvcc +- cuda-nvrtc-dev +- cuda-nvtx-dev +- cuda-profiler-api +- cuda-sanitizer-api +- cuda-version=12.9 +- cupy>=14.0.1,!=14.1.0 +- cxx-compiler +- cython>=3.2.2,<3.3.0a0 +- dask-cuda==26.10.*,>=0.0.0a0 +- dlpack>=0.8,<1.0 +- doxygen=1.9.1 +- fastavro>=0.22.9 +- flatbuffers==24.3.25 +- fsspec>=0.6.0 +- gcc_linux-64=14.* +- gdb +- hypothesis>=6.131.7 +- identify>=2.5.20 +- include-what-you-use==0.24.0 +- ipykernel +- ipython +- jupyter_client +- libcufile-dev +- libcurand-dev +- libkvikio==26.10.*,>=0.0.0a0 +- libnvcomp-dev==5.3.0.16 +- libnvjitlink-dev +- librapidsmpf==26.10.*,>=0.0.0a0 +- librdkafka<2.15.0a0 +- librmm==26.10.*,>=0.0.0a0 +- libucxx==0.52.*,>=0.0.0a0 +- make +- matplotlib-base<3.11 +- mmh3 +- moto>=4.0.8 +- msgpack-python +- myst-nb +- nanoarrow +- nbconvert +- nbformat +- nbsphinx +- ninja +- notebook +- numba-cuda>=0.22.2 +- numba>=0.60.0,<0.65.0 +- numexpr +- numpy>=2.0,<3.0 +- numpydoc +- nvidia-ml-py>=12 +- nvtx>=0.2.1 +- odfpy +- openpyxl +- packaging +- pandas>=3.0.0,<3.0.4a0 +- pandoc +- polars>=1.35,<1.43 +- pre-commit +- psutil +- pyarrow>=19.0.0,<24 +- pytables +- pytest-benchmark +- pytest-cases>=3.8.2 +- pytest-cov +- pytest-httpserver +- pytest-rerunfailures!=16.0.0 +- pytest-xdist +- pytest<9.1.0 +- python-calamine +- python-confluent-kafka +- python-xxhash +- python>=3.11 +- pytorch>=2.4.0 +- pyxlsb +- rapids-build-backend>=0.4.0,<0.5.0 +- rapids-dask-dependency==26.10.*,>=0.0.0a0 +- rapids-logger==0.3.* +- rapidsmpf==26.10.*,>=0.0.0a0 +- ray-default>=2.55.1 +- rich +- rmm==26.10.*,>=0.0.0a0 +- s3fs>=2022.3.0 +- scikit-build-core>=0.11.0 +- scipy +- sphinx-autobuild +- sphinx-copybutton +- sphinx-markdown-tables +- sphinx-remove-toctrees +- sphinx>=8.1.0 +- sphinxcontrib-websupport +- streamz +- structlog +- sysroot_linux-64==2.28 +- typing_extensions>=4.0.0 +- ucxx==0.52.*,>=0.0.0a0 +- xlrd +- xlsxwriter +- zlib>=1.2.13 +- zstandard +- pip: + - nvidia-sphinx-theme +name: all_cuda-129_arch-x86_64 diff --git a/conda/environments/all_cuda-133_arch-aarch64.yaml b/conda/environments/all_cuda-133_arch-aarch64.yaml new file mode 100644 index 00000000000..875088c0549 --- /dev/null +++ b/conda/environments/all_cuda-133_arch-aarch64.yaml @@ -0,0 +1,124 @@ +# This file is generated by `rapids-dependency-file-generator`. +# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. +channels: +- rapidsai-nightly +- rapidsai +- conda-forge +dependencies: +- aiobotocore>=2.2.0,<3 +- binutils +- boto3>=1.21.21 +- botocore>=1.24.21 +- breathe>=4.35.0 +- c-compiler +- cachetools +- clang-tools==20.1.8 +- clang==20.1.8 +- cmake>=4.0 +- cramjam +- croaring==4.4.2 +- cuda-bindings>=13.0.1,<14.0 +- cuda-cudart-dev +- cuda-cupti-dev +- cuda-nvcc +- cuda-nvrtc-dev +- cuda-nvtx-dev +- cuda-profiler-api +- cuda-sanitizer-api +- cuda-version=13.3 +- cupy>=14.0.1,!=14.1.0 +- cxx-compiler +- cython>=3.2.2,<3.3.0a0 +- dask-cuda==26.10.*,>=0.0.0a0 +- dlpack>=0.8,<1.0 +- doxygen=1.9.1 +- fastavro>=0.22.9 +- flatbuffers==24.3.25 +- fsspec>=0.6.0 +- gcc_linux-aarch64=14.* +- gdb +- hypothesis>=6.131.7 +- identify>=2.5.20 +- include-what-you-use==0.24.0 +- ipykernel +- ipython +- jupyter_client +- libcufile-dev +- libcurand-dev +- libkvikio==26.10.*,>=0.0.0a0 +- libnvcomp-dev==5.3.0.16 +- libnvjitlink-dev +- librapidsmpf==26.10.*,>=0.0.0a0 +- librdkafka<2.15.0a0 +- librmm==26.10.*,>=0.0.0a0 +- libucxx==0.52.*,>=0.0.0a0 +- make +- matplotlib-base<3.11 +- mmh3 +- moto>=4.0.8 +- msgpack-python +- myst-nb +- nanoarrow +- nbconvert +- nbformat +- nbsphinx +- ninja +- notebook +- numba-cuda>=0.22.2 +- numba>=0.60.0,<0.65.0 +- numexpr +- numpy>=2.0,<3.0 +- numpydoc +- nvidia-ml-py>=12 +- nvtx>=0.2.1 +- odfpy +- openpyxl +- packaging +- pandas>=3.0.0,<3.0.4a0 +- pandoc +- polars>=1.35,<1.43 +- pre-commit +- psutil +- pyarrow>=19.0.0,<24 +- pytables +- pytest-benchmark +- pytest-cases>=3.8.2 +- pytest-cov +- pytest-httpserver +- pytest-rerunfailures!=16.0.0 +- pytest-xdist +- pytest<9.1.0 +- python-calamine +- python-confluent-kafka +- python-xxhash +- python>=3.11 +- pytorch>=2.10.0 +- pyxlsb +- rapids-build-backend>=0.4.0,<0.5.0 +- rapids-dask-dependency==26.10.*,>=0.0.0a0 +- rapids-logger==0.3.* +- rapidsmpf==26.10.*,>=0.0.0a0 +- ray-default>=2.55.1 +- rich +- rmm==26.10.*,>=0.0.0a0 +- s3fs>=2022.3.0 +- scikit-build-core>=0.11.0 +- scipy +- sphinx-autobuild +- sphinx-copybutton +- sphinx-markdown-tables +- sphinx-remove-toctrees +- sphinx>=8.1.0 +- sphinxcontrib-websupport +- streamz +- structlog +- sysroot_linux-aarch64==2.28 +- typing_extensions>=4.0.0 +- ucxx==0.52.*,>=0.0.0a0 +- xlrd +- xlsxwriter +- zlib>=1.2.13 +- zstandard +- pip: + - nvidia-sphinx-theme +name: all_cuda-133_arch-aarch64 diff --git a/conda/environments/all_cuda-133_arch-x86_64.yaml b/conda/environments/all_cuda-133_arch-x86_64.yaml new file mode 100644 index 00000000000..fcb76af8b6b --- /dev/null +++ b/conda/environments/all_cuda-133_arch-x86_64.yaml @@ -0,0 +1,124 @@ +# This file is generated by `rapids-dependency-file-generator`. +# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. +channels: +- rapidsai-nightly +- rapidsai +- conda-forge +dependencies: +- aiobotocore>=2.2.0,<3 +- binutils +- boto3>=1.21.21 +- botocore>=1.24.21 +- breathe>=4.35.0 +- c-compiler +- cachetools +- clang-tools==20.1.8 +- clang==20.1.8 +- cmake>=4.0 +- cramjam +- croaring==4.4.2 +- cuda-bindings>=13.0.1,<14.0 +- cuda-cudart-dev +- cuda-cupti-dev +- cuda-nvcc +- cuda-nvrtc-dev +- cuda-nvtx-dev +- cuda-profiler-api +- cuda-sanitizer-api +- cuda-version=13.3 +- cupy>=14.0.1,!=14.1.0 +- cxx-compiler +- cython>=3.2.2,<3.3.0a0 +- dask-cuda==26.10.*,>=0.0.0a0 +- dlpack>=0.8,<1.0 +- doxygen=1.9.1 +- fastavro>=0.22.9 +- flatbuffers==24.3.25 +- fsspec>=0.6.0 +- gcc_linux-64=14.* +- gdb +- hypothesis>=6.131.7 +- identify>=2.5.20 +- include-what-you-use==0.24.0 +- ipykernel +- ipython +- jupyter_client +- libcufile-dev +- libcurand-dev +- libkvikio==26.10.*,>=0.0.0a0 +- libnvcomp-dev==5.3.0.16 +- libnvjitlink-dev +- librapidsmpf==26.10.*,>=0.0.0a0 +- librdkafka<2.15.0a0 +- librmm==26.10.*,>=0.0.0a0 +- libucxx==0.52.*,>=0.0.0a0 +- make +- matplotlib-base<3.11 +- mmh3 +- moto>=4.0.8 +- msgpack-python +- myst-nb +- nanoarrow +- nbconvert +- nbformat +- nbsphinx +- ninja +- notebook +- numba-cuda>=0.22.2 +- numba>=0.60.0,<0.65.0 +- numexpr +- numpy>=2.0,<3.0 +- numpydoc +- nvidia-ml-py>=12 +- nvtx>=0.2.1 +- odfpy +- openpyxl +- packaging +- pandas>=3.0.0,<3.0.4a0 +- pandoc +- polars>=1.35,<1.43 +- pre-commit +- psutil +- pyarrow>=19.0.0,<24 +- pytables +- pytest-benchmark +- pytest-cases>=3.8.2 +- pytest-cov +- pytest-httpserver +- pytest-rerunfailures!=16.0.0 +- pytest-xdist +- pytest<9.1.0 +- python-calamine +- python-confluent-kafka +- python-xxhash +- python>=3.11 +- pytorch>=2.10.0 +- pyxlsb +- rapids-build-backend>=0.4.0,<0.5.0 +- rapids-dask-dependency==26.10.*,>=0.0.0a0 +- rapids-logger==0.3.* +- rapidsmpf==26.10.*,>=0.0.0a0 +- ray-default>=2.55.1 +- rich +- rmm==26.10.*,>=0.0.0a0 +- s3fs>=2022.3.0 +- scikit-build-core>=0.11.0 +- scipy +- sphinx-autobuild +- sphinx-copybutton +- sphinx-markdown-tables +- sphinx-remove-toctrees +- sphinx>=8.1.0 +- sphinxcontrib-websupport +- streamz +- structlog +- sysroot_linux-64==2.28 +- typing_extensions>=4.0.0 +- ucxx==0.52.*,>=0.0.0a0 +- xlrd +- xlsxwriter +- zlib>=1.2.13 +- zstandard +- pip: + - nvidia-sphinx-theme +name: all_cuda-133_arch-x86_64 From f77d9ae11591e70ec983634c90acbb26ac38c4b9 Mon Sep 17 00:00:00 2001 From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Date: Fri, 14 Aug 2026 23:27:10 +0000 Subject: [PATCH 3/3] Add numba-cuda-mlir to conda env files --- conda/environments/all_cuda-129_arch-aarch64.yaml | 1 + conda/environments/all_cuda-129_arch-x86_64.yaml | 1 + conda/environments/all_cuda-133_arch-aarch64.yaml | 1 + conda/environments/all_cuda-133_arch-x86_64.yaml | 1 + 4 files changed, 4 insertions(+) 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