From b960e1e359a204bb75a9111c8b55fbacb6135fd5 Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Tue, 28 Mar 2023 15:50:17 -0500
Subject: [PATCH] Add librmm channel.

---
 ci/build_cpp.sh | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/ci/build_cpp.sh b/ci/build_cpp.sh
index 853ae095d3..856041d907 100755
--- a/ci/build_cpp.sh
+++ b/ci/build_cpp.sh
@@ -7,10 +7,16 @@ source rapids-env-update
 
 export CMAKE_GENERATOR=Ninja
 
+# TODO: Should we define the major version upstream?
+RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}"
+LIBRMM_CHANNEL=$(rapids-get-artifact ci/rmm/pull-request/1223/c3cf48a/rmm_conda_cpp_cuda${RAPIDS_CUDA_MAJOR}_$(arch).tar.gz)
+
 rapids-print-env
 
 rapids-logger "Begin cpp build"
 
-rapids-mamba-retry mambabuild conda/recipes/libraft
+rapids-mamba-retry mambabuild \
+    --channel "${LIBRMM_CHANNEL}" \
+    conda/recipes/libraft
 
 rapids-upload-conda-to-s3 cpp