From 30c547eaf985dff98ac018983317e2a98c7313cd Mon Sep 17 00:00:00 2001
From: Vyas Ramasubramani <vyasr@nvidia.com>
Date: Sun, 2 Feb 2025 06:44:20 +0000
Subject: [PATCH] librmm is now a runtime requirement of rmm

---
 dependencies.yaml         | 50 +++++++++++++++++++++++++--------------
 python/rmm/pyproject.toml |  1 +
 2 files changed, 33 insertions(+), 18 deletions(-)

diff --git a/dependencies.yaml b/dependencies.yaml
index 1d0aaa01f..a50d0f884 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -11,7 +11,7 @@ files:
       - checks
       - cuda
       - cuda_version
-      - depends_on_rapids-logger
+      - depends_on_rapids_logger
       - develop
       - docs
       - py_version
@@ -55,14 +55,14 @@ files:
       key: requires
     includes:
       - build
-      - depends_on_rapids-logger
+      - depends_on_rapids_logger
   py_cpp_run:
     output: pyproject
     pyproject_dir: python/librmm
     extras:
       table: project
     includes:
-      - depends_on_rapids-logger
+      - depends_on_rapids_logger
   py_build:
     output: pyproject
     pyproject_dir: python/rmm
@@ -79,6 +79,7 @@ files:
     includes:
       - build
       - cython_build
+      - depends_on_librmm
   py_run:
     output: pyproject
     pyproject_dir: python/rmm
@@ -86,6 +87,7 @@ files:
       table: project
     includes:
       - run
+      - depends_on_librmm
   py_optional_test:
     output: pyproject
     pyproject_dir: python/rmm
@@ -178,21 +180,33 @@ dependencies:
           - matrix: # All CUDA 11 versions
             packages:
               - &cuda_python11 cuda-python>=11.8.5,<12.0a0
+  depends_on_librmm:
+    common:
+      - output_types: conda
+        packages:
+          - &librmm_unsuffixed librmm==25.4.*,>=0.0.0a0
+      - output_types: requirements
+        packages:
+          # pip recognizes the index as a global option for the requirements.txt file
+          # This index is needed for librmm-cu{11,12}.
+          - --extra-index-url=https://pypi.nvidia.com
+          - --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple
+    specific:
       - output_types: [requirements, pyproject]
         matrices:
-            - matrix:
-                cuda: "12.*"
-                cuda_suffixed: "true"
-              packages:
-                - librmm-cu12==25.4.*,>=0.0.0a0
-            - matrix:
-                cuda: "11.*"
-                cuda_suffixed: "true"
-              packages:
-                - librmm-cu11==25.4.*,>=0.0.0a0
-            - matrix: null
-              packages:
-                - librmm==25.4.*,>=0.0.0a0
+          - matrix:
+              cuda: "12.*"
+              cuda_suffixed: "true"
+            packages:
+              - librmm-cu12==25.4.*,>=0.0.0a0
+          - matrix:
+              cuda: "11.*"
+              cuda_suffixed: "true"
+            packages:
+              - librmm-cu11==25.4.*,>=0.0.0a0
+          - matrix:
+            packages:
+              - *librmm_unsuffixed
   checks:
     common:
       - output_types: [conda, requirements]
@@ -345,7 +359,7 @@ dependencies:
               - numpy==1.23.*
           - matrix:
             packages:
-  depends_on_rapids-logger:
+  depends_on_rapids_logger:
     common:
       - output_types: [conda, requirements, pyproject]
         packages:
@@ -353,6 +367,6 @@ dependencies:
       - output_types: requirements
         packages:
           # pip recognizes the index as a global option for the requirements.txt file
-          # This index is needed for rapids-logger
+          # This index is needed for rapids_logger
           - --extra-index-url=https://pypi.nvidia.com
           - --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple
diff --git a/python/rmm/pyproject.toml b/python/rmm/pyproject.toml
index d457b0913..2771ee027 100644
--- a/python/rmm/pyproject.toml
+++ b/python/rmm/pyproject.toml
@@ -31,6 +31,7 @@ license = { text = "Apache 2.0" }
 requires-python = ">=3.10"
 dependencies = [
     "cuda-python>=11.8.5,<12.0a0",
+    "librmm==25.4.*,>=0.0.0a0",
     "numpy>=1.23,<3.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 classifiers = [