TEMP: use CI artifacts from RMM/dependency PRs for testing

bdice · bdice · commit 8fe66e5a686e · 2026-04-17T02:09:37.000-05:00
diff --git a/ci/build_cpp.sh b/ci/build_cpp.sh
@@ -1,10 +1,11 @@
 #!/bin/bash
 
-# SPDX-FileCopyrightText: Copyright (c) 2023-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 
 set -euo pipefail
+source ./ci/use_conda_packages_from_prs.sh
 
 source rapids-configure-sccache
 
diff --git a/ci/build_docs.sh b/ci/build_docs.sh
@@ -1,9 +1,10 @@
 #!/bin/bash
 
-# SPDX-FileCopyrightText: Copyright (c) 2023-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_conda_packages_from_prs.sh
 
 rapids-logger "Create test conda environment"
 . /opt/conda/etc/profile.d/conda.sh
diff --git a/ci/build_python.sh b/ci/build_python.sh
@@ -1,9 +1,10 @@
 #!/bin/bash
 
-# SPDX-FileCopyrightText: Copyright (c) 2023-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_conda_packages_from_prs.sh
 
 source rapids-configure-sccache
 
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -4,6 +4,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_wheels_from_prs.sh
 
 package_name=$1
 package_dir=$2
diff --git a/ci/build_wheel_cuopt.sh b/ci/build_wheel_cuopt.sh
@@ -5,6 +5,7 @@
 
 
 set -euo pipefail
+source ./ci/use_wheels_from_prs.sh
 
 source rapids-init-pip
 
diff --git a/ci/build_wheel_cuopt_mps_parser.sh b/ci/build_wheel_cuopt_mps_parser.sh
@@ -5,6 +5,7 @@
 
 
 set -euo pipefail
+source ./ci/use_wheels_from_prs.sh
 
 source rapids-init-pip
 
diff --git a/ci/build_wheel_cuopt_server.sh b/ci/build_wheel_cuopt_server.sh
@@ -4,6 +4,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_wheels_from_prs.sh
 
 source rapids-init-pip
 
diff --git a/ci/build_wheel_cuopt_sh_client.sh b/ci/build_wheel_cuopt_sh_client.sh
@@ -4,6 +4,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_wheels_from_prs.sh
 
 source rapids-init-pip
 
diff --git a/ci/build_wheel_libcuopt.sh b/ci/build_wheel_libcuopt.sh
@@ -3,6 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_wheels_from_prs.sh
 
 source rapids-init-pip
 
diff --git a/ci/test_cpp.sh b/ci/test_cpp.sh
@@ -1,9 +1,10 @@
 #!/bin/bash
 
-# SPDX-FileCopyrightText: Copyright (c) 2023-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_conda_packages_from_prs.sh
 
 . /opt/conda/etc/profile.d/conda.sh
 
diff --git a/ci/test_cpp_memcheck.sh b/ci/test_cpp_memcheck.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-# SPDX-FileCopyrightText: Copyright (c) 2023-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 if [[ "$(date +%A)" != "Friday" ]]; then
@@ -9,6 +9,7 @@ if [[ "$(date +%A)" != "Friday" ]]; then
 fi
 
 set -euo pipefail
+source ./ci/use_conda_packages_from_prs.sh
 
 . /opt/conda/etc/profile.d/conda.sh
 
diff --git a/ci/test_notebooks.sh b/ci/test_notebooks.sh
@@ -1,9 +1,10 @@
 #!/bin/bash
 
-# SPDX-FileCopyrightText: Copyright (c) 2023-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_conda_packages_from_prs.sh
 
 . /opt/conda/etc/profile.d/conda.sh
 
diff --git a/ci/test_python.sh b/ci/test_python.sh
@@ -4,6 +4,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_conda_packages_from_prs.sh
 
 . /opt/conda/etc/profile.d/conda.sh
 
diff --git a/ci/test_self_hosted_service.sh b/ci/test_self_hosted_service.sh
@@ -4,6 +4,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_wheels_from_prs.sh
 
 source rapids-init-pip
 
diff --git a/ci/test_skills_assets.sh b/ci/test_skills_assets.sh
@@ -9,6 +9,7 @@
 # CLI: run cuopt_cli on each sample .mps in API-CLI skill assets.
 
 set -euo pipefail
+source ./ci/use_conda_packages_from_prs.sh
 
 # Use rapids-logger in CI; fall back to echo for local testing
 if command -v rapids-logger &>/dev/null; then
diff --git a/ci/test_wheel_cuopt.sh b/ci/test_wheel_cuopt.sh
@@ -4,6 +4,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
+source ./ci/use_wheels_from_prs.sh
 
 # sets up a constraints file for 'pip' and puts its location in an exported variable PIP_EXPORT,
 # so those constraints will affect all future 'pip install' calls
diff --git a/ci/test_wheel_cuopt_server.sh b/ci/test_wheel_cuopt_server.sh
@@ -4,6 +4,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 set -eou pipefail
+source ./ci/use_wheels_from_prs.sh
 
 source rapids-init-pip
 
diff --git a/ci/use_conda_packages_from_prs.sh b/ci/use_conda_packages_from_prs.sh
@@ -0,0 +1,22 @@
+#!/bin/bash
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION.
+# SPDX-License-Identifier: Apache-2.0
+
+# download CI artifacts
+LIBRMM_CHANNEL=$(rapids-get-pr-artifact rmm 2361 cpp conda)
+RMM_CHANNEL=$(RAPIDS_PY_NOARCH_SUFFIX=abi3 rapids-get-pr-artifact rmm 2361 python conda)
+LIBRAFT_CHANNEL=$(rapids-get-pr-artifact raft 2996 cpp conda)
+RAFT_CHANNEL=$(RAPIDS_PY_NOARCH_SUFFIX=abi3 rapids-get-pr-artifact raft 2996 python conda)
+
+RAPIDS_PREPENDED_CONDA_CHANNELS=(
+    "${LIBRMM_CHANNEL}"
+    "${RMM_CHANNEL}"
+    "${LIBRAFT_CHANNEL}"
+    "${RAFT_CHANNEL}"
+)
+export RAPIDS_PREPENDED_CONDA_CHANNELS
+
+for _channel in "${RAPIDS_PREPENDED_CONDA_CHANNELS[@]}"
+do
+   conda config --system --add channels "${_channel}"
+done
diff --git a/ci/use_wheels_from_prs.sh b/ci/use_wheels_from_prs.sh
@@ -0,0 +1,27 @@
+#!/bin/bash
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION.
+# SPDX-License-Identifier: Apache-2.0
+
+source rapids-init-pip
+
+RAPIDS_PY_CUDA_SUFFIX=$(rapids-wheel-ctk-name-gen "${RAPIDS_CUDA_VERSION}")
+
+LIBRMM_WHEELHOUSE=$(
+  RAPIDS_PY_WHEEL_NAME="librmm_${RAPIDS_PY_CUDA_SUFFIX}" rapids-get-pr-artifact rmm 2361 cpp wheel
+)
+RMM_WHEELHOUSE=$(
+  RAPIDS_PY_WHEEL_NAME="rmm_${RAPIDS_PY_CUDA_SUFFIX}" rapids-get-pr-artifact rmm 2361 python wheel
+)
+LIBRAFT_WHEELHOUSE=$(
+  RAPIDS_PY_WHEEL_NAME="libraft_${RAPIDS_PY_CUDA_SUFFIX}" rapids-get-pr-artifact raft 2996 cpp wheel
+)
+RAFT_WHEELHOUSE=$(
+  RAPIDS_PY_WHEEL_NAME="pylibraft_${RAPIDS_PY_CUDA_SUFFIX}" rapids-get-pr-artifact raft 2996 python wheel
+)
+
+cat > "${PIP_CONSTRAINT}" <<EOF
+librmm-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo "${LIBRMM_WHEELHOUSE}"/librmm_*.whl)
+rmm-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo "${RMM_WHEELHOUSE}"/rmm_*.whl)
+libraft-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo "${LIBRAFT_WHEELHOUSE}"/libraft_*.whl)
+pylibraft-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo "${RAFT_WHEELHOUSE}"/pylibraft_*.whl)
+EOF

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@`
`5`	`5`
`6`	`6`
`7`	`7`	`set -euo pipefail`
	`8`	`+source ./ci/use_wheels_from_prs.sh`
`8`	`9`
`9`	`10`	`source rapids-init-pip`
`10`	`11`