WIP: [DO NOT MERGE] test libraft and libcugraph wheels

ci/build_cpp.sh

@@ -11,6 +11,8 @@ source rapids-date-string
 
 export CMAKE_GENERATOR=Ninja
 
+source ./ci/use_conda_packages_from_prs.sh
+
 rapids-print-env
 
 rapids-logger "Begin cpp build"

ci/build_docs.sh

@@ -14,6 +14,8 @@ rapids-dependency-file-generator \
   --file-key docs \
   --matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION}" | tee env.yaml
 
+source ./ci/use_conda_packages_from_prs.sh
+
 rapids-mamba-retry env create --yes -f env.yaml -n docs
 
 # Temporarily allow unbound variables for conda activation.

ci/build_python.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright (c) 2022-2024, NVIDIA CORPORATION.
+# Copyright (c) 2022-2025, NVIDIA CORPORATION.
 
 set -euo pipefail
 
@@ -11,6 +11,8 @@ source rapids-date-string
 
 export CMAKE_GENERATOR=Ninja
 
+source ./ci/use_conda_packages_from_prs.sh
+
 rapids-print-env
 
 CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)

ci/build_wheel_cugraph-dgl.sh

@@ -1,9 +1,11 @@
 #!/bin/bash
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# Copyright (c) 2024-2025, NVIDIA CORPORATION.
 
 set -euo pipefail
 
 package_dir="python/cugraph-dgl"
 
+source ./ci/use_wheels_from_prs.sh
+
 ./ci/build_wheel.sh cugraph-dgl ${package_dir}
 ./ci/validate_wheel.sh ${package_dir} dist

ci/build_wheel_cugraph-pyg.sh

@@ -1,9 +1,11 @@
 #!/bin/bash
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# Copyright (c) 2024-2025, NVIDIA CORPORATION.
 
 set -euo pipefail
 
 package_dir="python/cugraph-pyg"
 
+source ./ci/use_wheels_from_prs.sh
+
 ./ci/build_wheel.sh cugraph-pyg ${package_dir}
 ./ci/validate_wheel.sh ${package_dir} dist

ci/build_wheel_pylibwholegraph.sh

@@ -1,10 +1,12 @@
 #!/bin/bash
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# Copyright (c) 2024-2025, NVIDIA CORPORATION.
 
 set -euo pipefail
 
 package_dir="python/pylibwholegraph"
 
+source ./ci/use_wheels_from_prs.sh
+
 export SKBUILD_CMAKE_ARGS="-DDETECT_CONDA_ENV=OFF;-DBUILD_SHARED_LIBS=OFF;-DCMAKE_MESSAGE_LOG_LEVEL=VERBOSE;-DCUDA_STATIC_RUNTIME=ON;-DWHOLEGRAPH_BUILD_WHEELS=ON"
 
 ./ci/build_wheel.sh pylibwholegraph ${package_dir}

ci/test_cpp.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright (c) 2022-2024, NVIDIA CORPORATION.
+# Copyright (c) 2022-2025, NVIDIA CORPORATION.
 
 set -euo pipefail
 
@@ -10,6 +10,8 @@ cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../
 
 RAPIDS_VERSION="$(rapids-version)"
 
+source ./ci/use_conda_packages_from_prs.sh
+
 CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
 
 rapids-logger "Generate C++ testing dependencies"

ci/test_notebooks.sh

@@ -21,6 +21,8 @@ rapids-dependency-file-generator \
   --prepend-channel "${PYTHON_CHANNEL}" \
 | tee env.yaml
 
+source ./ci/use_conda_packages_from_prs.sh
+
 rapids-mamba-retry env create --yes -f env.yaml -n test
 
 # Temporarily allow unbound variables for conda activation.

ci/test_python.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright (c) 2022-2024, NVIDIA CORPORATION.
+# Copyright (c) 2022-2025, NVIDIA CORPORATION.
 
 set -euo pipefail
 
@@ -10,6 +10,8 @@ cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../
 
 RAPIDS_VERSION="$(rapids-version)"
 
+source ./ci/use_conda_packages_from_prs.sh
+
 rapids-logger "Downloading artifacts from previous jobs"
 CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
 PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python)

ci/test_wheel_cugraph-dgl.sh

@@ -1,10 +1,12 @@
 #!/bin/bash
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# Copyright (c) 2024-2025, NVIDIA CORPORATION.
 
 set -eoxu pipefail
 
 package_name="cugraph-dgl"
 
+source ./ci/use_wheels_from_prs.sh
+
 mkdir -p ./dist
 RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
 

ci/test_wheel_cugraph-pyg.sh

@@ -1,10 +1,12 @@
 #!/bin/bash
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# Copyright (c) 2024-2025, NVIDIA CORPORATION.
 
 set -eoxu pipefail
 
 package_name="cugraph-pyg"
 
+source ./ci/use_wheels_from_prs.sh
+
 mkdir -p ./dist
 RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
 

ci/test_wheel_pylibwholegraph.sh

@@ -1,10 +1,12 @@
 #!/bin/bash
-# Copyright (c) 2023-2024, NVIDIA CORPORATION.
+# Copyright (c) 2023-2025, NVIDIA CORPORATION.
 
 set -e          # abort the script on error
 set -o pipefail # piped commands propagate their error
 set -E          # ERR traps are inherited by subcommands
 
+source ./ci/use_wheels_from_prs.sh
+
 mkdir -p ./dist
 RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
 RAPIDS_PY_WHEEL_NAME="pylibwholegraph_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-from-s3 ./dist

ci/use_conda_packages_from_prs.sh

@@ -0,0 +1,9 @@
+# Copyright (c) 2025, NVIDIA CORPORATION.
+
+CUGRAPH_COMMIT="2dbd946375d83dc30d29f1b09a013e3a6159483d"
+
+CUGRAPH_CPP_CHANNEL=$(rapids-get-pr-conda-artifact cugraph 4804 cpp "${CUGRAPH_COMMIT:0:7}")
+CUGRAPH_PYTHON_CHANNEL=$(rapids-get-pr-conda-artifact cugraph 4804 python "${CUGRAPH_COMMIT:0:7}")
+
+conda config --system --add channels "${CUGRAPH_CPP_CHANNEL}"
+conda config --system --add channels "${CUGRAPH_PYTHON_CHANNEL}"

ci/use_wheels_from_prs.sh

@@ -0,0 +1,22 @@
+# Copyright (c) 2025, NVIDIA CORPORATION.
+
+RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
+
+CUGRAPH_COMMIT="2dbd946375d83dc30d29f1b09a013e3a6159483d"
+CUGRAPH_CHANNEL=$(
+  RAPIDS_PY_WHEEL_NAME="cugraph_${RAPIDS_PY_CUDA_SUFFIX}" rapids-get-pr-wheel-artifact cugraph 4804 python "${CUGRAPH_COMMIT:0:}"
+)
+LIBCUGRAPH_CHANNEL=$(
+  RAPIDS_PY_WHEEL_NAME="libcugraph_${RAPIDS_PY_CUDA_SUFFIX}" rapids-get-pr-wheel-artifact cugraph 4804 cpp "${CUGRAPH_COMMIT:0:7}"
+)
+PYLIBCUGRAPH_CHANNEL=$(
+  RAPIDS_PY_WHEEL_NAME="pylibcugraph_${RAPIDS_PY_CUDA_SUFFIX}" rapids-get-pr-wheel-artifact cugraph 4804 python "${CUGRAPH_COMMIT:0:}"
+)
+
+cat > ./constraints.txt <<EOF
+cugraph-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${CUGRAPH_CHANNEL}/cugraph_*.whl)
+libcugraph-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${LIBCUGRAPH_CHANNEL}/libcugraph_*.whl)
+pylibcugraph-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${PYLIBCUGRAPH_CHANNEL}/pylibcugraph_*.whl)
+EOF
+
+export PIP_CONSTRAINT=$(pwd)/constraints.txt