Add back librmm to Conda env

Author: chyunsu3

ci/build_wheel_libnvforest.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION.
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
@@ -38,6 +38,7 @@ EXCLUDE_ARGS=(
   --exclude "libcusparse.so.*"
   --exclude "libnvJitLink.so.*"
   --exclude "librapids_logger.so"
+  --exclude "librmm.so"
 )
 
 export SKBUILD_CMAKE_ARGS="-DDISABLE_DEPRECATION_WARNINGS=ON"

ci/build_wheel_nvforest.sh

@@ -30,6 +30,7 @@ EXCLUDE_ARGS=(
   --exclude "libcusparse.so.*"
   --exclude "libnvJitLink.so.*"
   --exclude "librapids_logger.so"
+  --exclude "librmm.so"
   --exclude "libtreelite.so"
 )
 

ci/release/update-version.sh

@@ -111,6 +111,7 @@ DEPENDENCIES=(
   libnvforest-tests
   libraft
   libraft-headers
+  librmm
   pylibraft
 )
 for DEP in "${DEPENDENCIES[@]}"; do

conda/environments/all_cuda-129_arch-aarch64.yaml

@@ -23,6 +23,7 @@ dependencies:
 - gcc_linux-aarch64=14.*
 - graphviz
 - libraft==26.12.*,>=0.0.0a0
+- librmm==26.12.*,>=0.0.0a0
 - ninja
 - numpy>=2.0,<3.0a0
 - numpydoc

conda/environments/all_cuda-129_arch-x86_64.yaml

@@ -23,6 +23,7 @@ dependencies:
 - gcc_linux-64=14.*
 - graphviz
 - libraft==26.12.*,>=0.0.0a0
+- librmm==26.12.*,>=0.0.0a0
 - ninja
 - numpy>=2.0,<3.0a0
 - numpydoc

conda/environments/all_cuda-133_arch-aarch64.yaml

@@ -23,6 +23,7 @@ dependencies:
 - gcc_linux-aarch64=14.*
 - graphviz
 - libraft==26.12.*,>=0.0.0a0
+- librmm==26.12.*,>=0.0.0a0
 - ninja
 - numpy>=2.0,<3.0a0
 - numpydoc

conda/environments/all_cuda-133_arch-x86_64.yaml

@@ -23,6 +23,7 @@ dependencies:
 - gcc_linux-64=14.*
 - graphviz
 - libraft==26.12.*,>=0.0.0a0
+- librmm==26.12.*,>=0.0.0a0
 - ninja
 - numpy>=2.0,<3.0a0
 - numpydoc

conda/environments/clang_tidy_cuda-129_arch-x86_64.yaml

@@ -16,6 +16,7 @@ dependencies:
 - cxx-compiler
 - gcc_linux-64=14.*
 - libraft-headers==26.12.*,>=0.0.0a0
+- librmm==26.12.*,>=0.0.0a0
 - llvm-openmp==15.0.7
 - ninja
 - sysroot_linux-64==2.28

conda/environments/clang_tidy_cuda-133_arch-x86_64.yaml

@@ -16,6 +16,7 @@ dependencies:
 - cxx-compiler
 - gcc_linux-64=14.*
 - libraft-headers==26.12.*,>=0.0.0a0
+- librmm==26.12.*,>=0.0.0a0
 - llvm-openmp==15.0.7
 - ninja
 - sysroot_linux-64==2.28

conda/environments/cpp_all_cuda-129_arch-x86_64.yaml

@@ -14,6 +14,7 @@ dependencies:
 - cxx-compiler
 - gcc_linux-64=14.*
 - libraft-headers==26.12.*,>=0.0.0a0
+- librmm==26.12.*,>=0.0.0a0
 - ninja
 - sysroot_linux-64==2.28
 name: cpp_all_cuda-129_arch-x86_64