Add librmm channel.

rapidsai · Mar 28, 2023 · b960e1e · b960e1e
1 parent 7f3b1ac
commit b960e1e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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