Depend on librmm wheel

Contributes to rapidsai/build-planning#33
KyleFromNVIDIA · Jul 23, 2024 · 4ddcc9b · 4ddcc9b
1 parent 67ec8f8
commit 4ddcc9b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -206,14 +206,17 @@ dependencies:
         matrices:
           - matrix: {cuda: "12.*"}
             packages:
+              - librmm-cu12==24.10.*,>=0.0.0a0
               - pylibraft-cu12==24.10.*,>=0.0.0a0
               - rmm-cu12==24.10.*,>=0.0.0a0
           - matrix: {cuda: "11.*"}
             packages:
+              - &librmm_cu11 librmm-cu11==24.10.*,>=0.0.0a0
               - &pylibraft_cu11 pylibraft-cu11==24.10.*,>=0.0.0a0
               - &rmm_cu11 rmm-cu11==24.10.*,>=0.0.0a0
           - matrix:
             packages:
+              - librmm==24.10.*,>=0.0.0a0
               - pylibraft==24.10.*,>=0.0.0a0
               - rmm==24.10.*,>=0.0.0a0
 

diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml
@@ -157,6 +157,7 @@ requires = [
     "cmake>=3.26.4,!=3.30.0",
     "cuda-python",
     "cython>=3.0.0",
+    "librmm==24.10.*,>=0.0.0a0",
     "ninja",
     "pylibraft==24.10.*,>=0.0.0a0",
     "rmm==24.10.*,>=0.0.0a0",