From 689fe923ee649dae725670a9903a9f1b9fb3ed90 Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Fri, 21 Feb 2025 20:37:57 -0600
Subject: [PATCH] Recipe revisions.

---
 conda/recipes/librmm/recipe.yaml | 40 +++++++++++++++++++++++++-------
 1 file changed, 32 insertions(+), 8 deletions(-)

diff --git a/conda/recipes/librmm/recipe.yaml b/conda/recipes/librmm/recipe.yaml
index 580cd6efb..d0efe586d 100644
--- a/conda/recipes/librmm/recipe.yaml
+++ b/conda/recipes/librmm/recipe.yaml
@@ -56,8 +56,9 @@ outputs:
       script:
         - cmake --install build
     requirements:
-      host:
+      build:
         - cmake ${{ cmake_version }}
+      host:
         - cuda-version =${{ cuda_version }}
       run:
         - ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }}
@@ -67,8 +68,8 @@ outputs:
       run_exports:
         - ${{ pin_subpackage("librmm", upper_bound="x.x") }}
       ignore_run_exports:
-        from_package:
-          - ${{ compiler("cuda") }}
+        by_name:
+          - cuda-version
     tests:
       - script:
           - "test -d \"${PREFIX}/include/rmm\""
@@ -84,8 +85,9 @@ outputs:
       script:
         - cmake --install build --component testing
     requirements:
-      host:
+      build:
         - cmake ${{ cmake_version }}
+      host:
         - cuda-version =${{ cuda_version }}
         - if: cuda_major == "11"
           then: cudatoolkit
@@ -98,9 +100,10 @@ outputs:
         - ${{ pin_subpackage("librmm", exact=True) }}
       ignore_run_exports:
         from_package:
-          - ${{ compiler("cuda") }}
-          - if: cuda_major == "11"
+          - if: cuda_major != "11"
             then: cuda-cudart-dev
+        by_name:
+          - cuda-version
     about:
       homepage: ${{ load_from_file("python/librmm/pyproject.toml").project.urls.Homepage }}
       license: ${{ load_from_file("python/librmm/pyproject.toml").project.license.text | replace(" ", "-") }}
@@ -111,8 +114,28 @@ outputs:
     build:
       string: cuda${{ cuda_major }}_${{ date_string }}_${{ head_rev }}
       script:
-        - ./examples/build.sh --install
+        content:
+          - ./examples/build.sh --install
+        env:
+          CMAKE_C_COMPILER_LAUNCHER: ${{ env.get("CMAKE_C_COMPILER_LAUNCHER") }}
+          CMAKE_CUDA_COMPILER_LAUNCHER: ${{ env.get("CMAKE_CUDA_COMPILER_LAUNCHER") }}
+          CMAKE_CXX_COMPILER_LAUNCHER: ${{ env.get("CMAKE_CXX_COMPILER_LAUNCHER") }}
+          CMAKE_GENERATOR: ${{ env.get("CMAKE_GENERATOR") }}
+          PARALLEL_LEVEL: ${{ env.get("PARALLEL_LEVEL") }}
+          SCCACHE_BUCKET: ${{ env.get("SCCACHE_BUCKET") }}
+          SCCACHE_IDLE_TIMEOUT: ${{ env.get("SCCACHE_IDLE_TIMEOUT") }}
+          SCCACHE_REGION: ${{ env.get("SCCACHE_REGION") }}
+          SCCACHE_S3_USE_SSL: ${{ env.get("SCCACHE_S3_USE_SSL") }}
+          SCCACHE_S3_NO_CREDENTIALS: ${{ env.get("SCCACHE_S3_NO_CREDENTIALS") }}
+          SCCACHE_S3_KEY_PREFIX: librmm-${{ env.get("RAPIDS_CONDA_ARCH") }}
     requirements:
+      build:
+        - cmake ${{ cmake_version }}
+        - ninja
+        - ${{ compiler("c") }}
+        - ${{ compiler("cxx") }}
+        - ${{ compiler("cuda") }}
+        - cuda-version =${{ cuda_version }}
       host:
         - cuda-version =${{ cuda_version }}
         - if: cuda_major == "11"
@@ -127,10 +150,11 @@ outputs:
         - ${{ pin_subpackage("librmm", exact=True) }}
       ignore_run_exports:
         from_package:
-          - if: cuda_major == "11"
+          - if: cuda_major != "11"
             then: cuda-cudart-dev
         by_name:
           - cuda-version
+          - librmm
     about:
       homepage: ${{ load_from_file("python/librmm/pyproject.toml").project.urls.Homepage }}
       license: ${{ load_from_file("python/librmm/pyproject.toml").project.license.text | replace(" ", "-") }}