From 7f3b1acc5658347e9079a636eeefbf7feecc851b Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Tue, 28 Mar 2023 15:49:43 -0500
Subject: [PATCH] Add librmm cuda pinning.

---
 conda/recipes/libraft/meta.yaml | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/conda/recipes/libraft/meta.yaml b/conda/recipes/libraft/meta.yaml
index 5f7ffa1d93..058d216042 100644
--- a/conda/recipes/libraft/meta.yaml
+++ b/conda/recipes/libraft/meta.yaml
@@ -52,6 +52,9 @@ outputs:
         - {{ compiler('cuda11') }} {{ cuda_version }}
         {% else %}
         - {{ compiler('cuda') }} {{ cuda_version }}
+        - cuda-cudart ={{ cuda_version }}
+        - cuda-cudart-dev ={{ cuda_version }}
+        - cuda-cudart-static ={{ cuda_version }}
         {% endif %}
         - cmake {{ cmake_version }}
         - ninja
@@ -80,7 +83,7 @@ outputs:
         - libcusparse {{ libcusparse_host_version }}
         - libcusparse-dev {{ libcusparse_host_version }}
         {% endif %}
-        - librmm ={{ minor_version }}
+        - librmm ={{ minor_version }}*=cuda{{ cuda_major }}
       run:
         {% if cuda_major == "11" %}
         - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }}
@@ -105,7 +108,7 @@ outputs:
         - libcusparse {{ libcusparse_run_version }}
         - libcusparse-dev {{ libcusparse_run_version }}
         {% endif %}
-        - librmm ={{ minor_version }}
+        - librmm ={{ minor_version }}*=cuda{{ cuda_major }}
     about:
       home: https://rapids.ai/
       license: Apache-2.0