dependencies.yaml

# SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

# Dependency list for https://github.com/rapidsai/dependency-file-generator
files:
  all:
    output: conda
    matrix:
      cuda: ["12.9", "13.4"]
      arch: [x86_64, aarch64]
    includes:
      - build
      - cython_build
      - checks
      - cuda_version
      - depends_on_rapids_logger
      - develop
      - docs
      - py_version
      - rapids_build_skbuild
      - run
      - test_python
  test_python:
    output: none
    includes:
      - cuda_version
      - py_version
      - test_python
      - depends_on_librmm
      - depends_on_rmm
  test_pytorch:
    output: none
    includes:
      - cuda_version
      - py_version
      - test_python
      - depends_on_pytorch
      - depends_on_librmm
      - depends_on_rmm
  test_cupy:
    output: none
    includes:
      - cuda_version
      - py_version
      - test_python
      - depends_on_cupy
      - depends_on_librmm
      - depends_on_rmm
  test_wheels_cupy:
    output: none
    includes:
      - depends_on_cupy
  test_cpp:
    output: none
    includes:
      - cuda_version
      - test_cpp
      - depends_on_librmm
      - depends_on_librmm_tests
      - depends_on_librmm_example
  test_cmake:
    output: none
    includes:
      - build
      - cuda_static
      - cuda_version
      - depends_on_rapids_logger
  checks:
    output: none
    includes:
      - checks
      - py_version
  docs:
    output: none
    includes:
      - cuda_version
      - docs
      - py_version
      - depends_on_rmm
      - depends_on_librmm
  py_cpp_build:
    output: pyproject
    pyproject_dir: python/librmm
    extras:
      table: build-system
    includes:
      - rapids_build_skbuild
  py_cpp_rapids_build:
    output: pyproject
    pyproject_dir: python/librmm
    extras:
      table: tool.rapids-build-backend
      key: requires
    includes:
      - build
      - depends_on_rapids_logger
  py_cpp_run:
    output: pyproject
    pyproject_dir: python/librmm
    extras:
      table: project
    includes:
      - depends_on_rapids_logger
  py_build:
    output: pyproject
    pyproject_dir: python/rmm
    extras:
      table: build-system
    includes:
      - rapids_build_skbuild
  py_rapids_build:
    output: pyproject
    pyproject_dir: python/rmm
    matrix:
      cuda_suffixed: ["false"]
      use_cuda_wheels: ["true"]
    extras:
      table: tool.rapids-build-backend
      key: requires
    includes:
      - build
      - cython_build
      - depends_on_librmm
  py_run:
    output: pyproject
    pyproject_dir: python/rmm
    matrix:
      cuda_suffixed: ["false"]
      use_cuda_wheels: ["true"]
    extras:
      table: project
    includes:
      - run
      - depends_on_librmm
  py_optional_test:
    output: pyproject
    pyproject_dir: python/rmm
    matrix:
      cuda_suffixed: ["false"]
      use_cuda_wheels: ["true"]
    extras:
      table: project.optional-dependencies
      key: test
    includes:
      - test_python
  torch_only:
    output: none
    includes:
      - depends_on_pytorch
channels:
  - rapidsai-nightly
  - rapidsai
  - conda-forge
dependencies:
  rapids_build_skbuild:
    common:
      - output_types: [conda, requirements, constraints, pyproject]
        packages:
          - rapids-build-backend>=0.4.0,<0.5.0
      - output_types: [conda, constraints]
        packages:
          - scikit-build-core>=0.11.0
      - output_types: [requirements, pyproject]
        packages:
          - scikit-build-core[pyproject]>=0.11.0
  build:
    common:
      - output_types: [conda, requirements, constraints, pyproject]
        packages:
          - &cmake_ver cmake>=4.0
          - ninja
      - output_types: conda
        packages:
          - c-compiler
          - cxx-compiler
          - cuda-nvcc
    specific:
      - output_types: conda
        matrices:
          - matrix:
              arch: x86_64
            packages:
              - gcc_linux-64=14.*
              - sysroot_linux-64==2.28
          - matrix:
              arch: aarch64
            packages:
              - gcc_linux-aarch64=14.*
              - sysroot_linux-aarch64==2.28
  cython_build:
    common:
      - output_types: [conda, requirements, constraints, pyproject]
        packages:
          - cython>=3.2.2
    specific:
      - output_types: [conda, requirements, constraints, pyproject]
        matrices:
          - matrix:
              cuda: "12.*"
            packages:
              - &cuda_bindings12 cuda-bindings>=12.9.3,<13.0
          - matrix: # All CUDA 13 versions
            packages:
              - &cuda_bindings13 cuda-bindings>=13.0.2,<14.0
  depends_on_librmm:
    common:
      - output_types: conda
        packages:
          - &librmm_unsuffixed librmm==26.12.*,>=0.0.0a0
    specific:
      - output_types: [requirements, constraints, pyproject]
        matrices:
          - matrix:
              cuda: "12.*"
              cuda_suffixed: "true"
            packages:
              - librmm-cu12==26.12.*,>=0.0.0a0
          - matrix:
              cuda: "13.*"
              cuda_suffixed: "true"
            packages:
              - librmm-cu13==26.12.*,>=0.0.0a0
          - matrix:
              cuda_suffixed: "false"
            packages:
              - *librmm_unsuffixed
  depends_on_rmm:
    common:
      - output_types: conda
        packages:
          - rmm==26.12.*,>=0.0.0a0
  depends_on_librmm_tests:
    common:
      - output_types: conda
        packages:
          - librmm-tests==26.12.*,>=0.0.0a0
  depends_on_librmm_example:
    common:
      - output_types: conda
        packages:
          - librmm-example==26.12.*,>=0.0.0a0
  checks:
    common:
      - output_types: [conda, requirements, constraints]
        packages:
          - pre-commit
          # pre-commit requires identify minimum version 1.0, but clang-format
          # requires textproto support and that was added in 2.5.20, so we need
          # to call out the minimum version needed for our plugins
          - identify>=2.5.20
      - output_types: conda
        packages:
          - &doxygen doxygen=1.18.0
  cuda_static:
    common:
      - output_types: conda
        packages:
          - cuda-cudart-static
  # 'cuda_version' intentionally does not contain fallback entries... we want
  # a loud error if an unsupported 'cuda' value is passed
  cuda_version:
    specific:
      - output_types: conda
        matrices:
          - matrix:
              cuda: "12.2"
            packages:
              - cuda-version=12.2
          - matrix:
              cuda: "12.5"
            packages:
              - cuda-version=12.5
          - matrix:
              cuda: "12.8"
            packages:
              - cuda-version=12.8
          - matrix:
              cuda: "12.9"
            packages:
              - cuda-version=12.9
          - matrix:
              cuda: "13.0"
            packages:
              - cuda-version=13.0
          - matrix:
              cuda: "13.1"
            packages:
              - cuda-version=13.1
          - matrix:
              cuda: "13.2"
            packages:
              - cuda-version=13.2
          - matrix:
              cuda: "13.3"
            packages:
              - cuda-version=13.3
          - matrix:
              cuda: "13.4"
            packages:
              - cuda-version=13.4
      - output_types: [requirements, constraints]
        matrices:
          # if use_cuda_wheels=false is provided, do not add dependencies on any CUDA wheels
          # (e.g. for DLFW and pip devcontainers)
          - matrix:
              use_cuda_wheels: "false"
            packages:
          - matrix:
              arch: aarch64
              cuda: "12.2"
              use_cuda_wheels: "true"
            packages:
              # some components (like nvidia-cublas-cu12 and nvidia-cuda-nvcc-cu12) didn't have
              # aarch64 wheels until CTK 12.3, so allow a slightly looser bound here
              - cuda-toolkit>=12.2,<12.4
          - matrix:
              cuda: "12.2"
              use_cuda_wheels: "true"
            packages:
              - cuda-toolkit==12.2.*
          - matrix:
              cuda: "12.5"
              use_cuda_wheels: "true"
            packages:
              - cuda-toolkit==12.5.*
          - matrix:
              cuda: "12.8"
              use_cuda_wheels: "true"
            packages:
              - cuda-toolkit==12.8.*
          - matrix:
              cuda: "12.9"
              use_cuda_wheels: "true"
            packages:
              - cuda-toolkit==12.9.*
          - matrix:
              cuda: "13.0"
              use_cuda_wheels: "true"
            packages:
              - cuda-toolkit==13.0.*
          - matrix:
              cuda: "13.1"
              use_cuda_wheels: "true"
            packages:
              - cuda-toolkit==13.1.*
          - matrix:
              cuda: "13.2"
              use_cuda_wheels: "true"
            packages:
              - cuda-toolkit==13.2.*
          - matrix:
              cuda: "13.3"
              use_cuda_wheels: "true"
            packages:
              - cuda-toolkit==13.3.*
          - matrix:
              cuda: "13.4"
              use_cuda_wheels: "true"
            packages:
              - cuda-toolkit==13.4.*
  develop:
    common:
      - output_types: conda
        packages:
          - clang==20.1.8
          - clang-tools==20.1.8
          - gcovr>=5.0
  docs:
    common:
      - output_types: conda
        packages:
          - breathe>=4.35.0
          - *doxygen
          - ipython
          - make
          - myst-parser
          - nbsphinx
          - &numba numba>=0.60.0,<0.65.0
          - &numba_cuda numba-cuda>=0.30.2
          - numpydoc
          - sphinx
          - sphinx-copybutton
          - sphinx-markdown-tables
          - sphinxcontrib-jquery
          - pip:
              - nvidia-sphinx-theme
  py_version:
    specific:
      - output_types: conda
        matrices:
          - matrix:
              py: "3.12"
            packages:
              - python=3.12
          - matrix:
              py: "3.13"
            packages:
              - python=3.13
          - matrix:
              py: "3.14"
            packages:
              - python=3.14
          - matrix:
            packages:
              - python>=3.12
  run:
    common:
      - output_types: [conda, requirements, constraints, pyproject]
        packages:
          - numpy>=2.0,<3.0
    specific:
      - output_types: [conda, requirements, constraints, pyproject]
        matrices:
          - matrix:
              cuda: "12.*"
            packages:
              - *cuda_bindings12
          - matrix: # All CUDA 13 versions
            packages:
              - *cuda_bindings13
  test_cpp:
    common:
      - output_types: conda
        packages:
          - *cmake_ver
  test_python:
    common:
      - output_types: [conda, requirements, constraints, pyproject]
        packages:
          - cuda-core>=1.0,<2
          - packaging
          - pytest
          - pytest-cov
    specific:
      - output_types: [conda, requirements, constraints, pyproject]
        matrices:
          - matrix:
              dependencies: "oldest"
              cuda: "12.*"
            packages:
              # Older bindings segfault when cuda-core releases contexts at shutdown.
              - cuda-bindings==12.9.3
          - matrix:
              dependencies: "oldest"
              cuda: "13.*"
            packages:
              # Older bindings segfault when cuda-core releases contexts at shutdown.
              - cuda-bindings==13.0.2
          - matrix:
            packages:
      - output_types: [conda, requirements, constraints, pyproject]
        matrices:
          - matrix:
              dependencies: "oldest"
            packages:
              # cuda-core 1.0.0 cannot import with the oldest CUDA 13 bindings.
              - cuda-core==1.0.1
              - numba==0.60.0
              - numpy==2.0.*
          - matrix:
            packages:
              - *numba
      - output_types: [conda]
        matrices:
          - matrix:
              dependencies: "oldest"
            packages:
              # Earlier numba-cuda releases require cuda-core<1.0.0.
              - &numba_cuda_oldest numba-cuda==0.30.2
          - matrix:
            packages:
              - *numba_cuda
      - output_types: constraints
        matrices:
          - matrix:
              dependencies: "oldest"
            packages:
              # Constraint must not have extras
              - *numba_cuda_oldest
          - matrix:
            packages:
              - *numba_cuda
      - output_types: [requirements, pyproject]
        matrices:
          - matrix:
              cuda: "12.*"
              cuda_suffixed: "true"
            packages:
              - numba-cuda[cu12]>=0.30.2
          - matrix:
              cuda: "13.*"
              cuda_suffixed: "true"
            packages:
              - numba-cuda[cu13]>=0.30.2
          # fallback to numba-cuda with no extra CUDA packages if 'cuda_suffixed' isn't true
          - matrix:
            packages:
              - *numba_cuda

  depends_on_rapids_logger:
    common:
      - output_types: [conda, requirements, constraints, pyproject]
        packages:
          - rapids-logger==0.3.*
  depends_on_pytorch:
    specific:
      - output_types: conda
        matrices:
          - matrix:
              require_gpu: "true"
            packages:
              - pytorch-gpu>=2.9
          - matrix:
            packages:
              - pytorch>=2.9
      # The 'pytorch.org' indices referenced in --extra-index-url below host CPU-only variants too,
      # so requirements like '>=' are not safe.
      #
      # Using '==' and a version with the CUDA specifier like '+cu130' is the most reliable way to ensure
      # the packages we want are pulled (at the expense of needing to maintain this list).
      #
      # 'torch' tightly pins wheels to a single {major}.{minor} CTK version.
      #
      # This list only contains entries exactly matching CUDA {major}.{minor} that we test in RMM CI,
      # to ensure a loud error alerts us to the need to update this list (or CI scripts) when new
      # CTKs are added to the support matrix.
      - output_types: [requirements, constraints]
        matrices:
          # avoid pulling in 'torch' in places like DLFW builds that prefer to install it other ways
          - matrix:
              no_pytorch: "true"
            packages:
          # matrices below ensure CUDA 'torch' packages are used
          - matrix:
              cuda: "12.9"
              dependencies: "oldest"
              require_gpu: "true"
            packages:
              - &torch_cu129_index --extra-index-url=https://download.pytorch.org/whl/cu129
              - torch==2.9.0+cu129
          - matrix:
              cuda: "12.9"
              require_gpu: "true"
            packages:
              - *torch_cu129_index
              - torch==2.13.0+cu129
          - matrix:
              cuda: "13.0"
              dependencies: "oldest"
              require_gpu: "true"
            packages:
              - &torch_cu130_index --extra-index-url=https://download.pytorch.org/whl/cu130
              - torch==2.9.0+cu130
          - matrix:
              cuda: "13.0"
              require_gpu: "true"
            packages:
              - *torch_cu130_index
              - torch==2.14.0+cu130
          - matrix:
              cuda: "13.1"
              require_gpu: "true"
            packages:
              - *torch_cu130_index
              - torch==2.13.0+cu130
          - matrix:
              cuda: "13.*"
              require_gpu: "true"
            packages:
              - --extra-index-url=https://download.pytorch.org/whl/cu132
              - torch==2.14.0+cu132
          - matrix:
              require_gpu: "false"
            packages:
              - torch>=2.9
  depends_on_cupy:
    common:
      - output_types: conda
        packages:
          - cupy>=14.0.1,!=14.1.0
    specific:
      - output_types: constraints
        matrices:
          - matrix:
              cuda: "12.*"
            packages:
              - &cupy_cu12 cupy-cuda12x>=14.0.1,!=14.1.0
          - matrix:
            packages:
              - &cupy_cu13 cupy-cuda13x>=14.0.1,!=14.1.0
      - output_types: [requirements, pyproject]
        matrices:
          - matrix:
              cuda: "12.*"
              use_cuda_wheels: "false"
            packages:
              - *cupy_cu12
          - matrix:
              cuda: "13.*"
              use_cuda_wheels: "false"
            packages:
              - *cupy_cu13
          - matrix:
              cuda: "12.*"
              use_cuda_wheels: "true"
            packages:
              - cupy-cuda12x[ctk]>=14.0.1,!=14.1.0
          # fallback to CUDA 13 versions if 'cuda' is '13.*' or not provided
          - matrix:
              use_cuda_wheels: "true"
            packages:
              - cupy-cuda13x[ctk]>=14.0.1,!=14.1.0