use rapids-get-pr-conda-artifact kvikio 479 cpp

rapidsai · Oct 3, 2024 · 4f94346 · 4f94346
1 parent 7cee919
commit 4f94346
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 0 deletions.
diff --git a/ci/build_cpp.sh b/ci/build_cpp.sh
@@ -15,8 +15,11 @@ rapids-print-env
 
 rapids-logger "Begin cpp build"
 
+LIBKVIKIO_CHANNEL=$(rapids-get-pr-conda-artifact kvikio 479 cpp)
+
 # With boa installed conda build forward to boa
 RAPIDS_PACKAGE_VERSION=$(rapids-generate-version) rapids-conda-retry mambabuild \
+    --channel "${LIBKVIKIO_CHANNEL}" \
     conda/recipes/libcudf
 
 rapids-upload-conda-to-s3 cpp
diff --git a/ci/build_docs.sh b/ci/build_docs.sh
@@ -25,10 +25,12 @@ rapids-print-env
 rapids-logger "Downloading artifacts from previous jobs"
 CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
 PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python)
+LIBKVIKIO_CHANNEL=$(rapids-get-pr-conda-artifact kvikio 479 cpp)
 
 rapids-mamba-retry install \
   --channel "${CPP_CHANNEL}" \
   --channel "${PYTHON_CHANNEL}" \
+  --channel "${LIBKVIKIO_CHANNEL}" \
   libcudf pylibcudf cudf dask-cudf
 
 export RAPIDS_DOCS_DIR="$(mktemp -d)"

diff --git a/ci/build_python.sh b/ci/build_python.sh
@@ -18,6 +18,7 @@ rapids-generate-version > ./VERSION
 rapids-logger "Begin py build"
 
 CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
+LIBKVIKIO_CHANNEL=$(rapids-get-pr-conda-artifact kvikio 479 cpp)
 
 # TODO: Remove `--no-test` flag once importing on a CPU
 # node works correctly
@@ -32,6 +33,7 @@ RAPIDS_PACKAGE_VERSION=$(head -1 ./VERSION) rapids-conda-retry mambabuild \
   --no-test \
   --channel "${CPP_CHANNEL}" \
   --channel "${RAPIDS_CONDA_BLD_OUTPUT_DIR}" \
+  --channel "${LIBKVIKIO_CHANNEL}" \
   conda/recipes/cudf
 
 RAPIDS_PACKAGE_VERSION=$(head -1 ./VERSION) rapids-conda-retry mambabuild \

diff --git a/ci/test_cpp_common.sh b/ci/test_cpp_common.sh
@@ -22,6 +22,7 @@ conda activate test
 set -u
 
 CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
+LIBKVIKIO_CHANNEL=$(rapids-get-pr-conda-artifact kvikio 479 cpp)
 
 RESULTS_DIR=${RAPIDS_TESTS_DIR:-"$(mktemp -d)"}
 RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${RESULTS_DIR}/test-results"}/
@@ -31,6 +32,7 @@ rapids-print-env
 
 rapids-mamba-retry install \
   --channel "${CPP_CHANNEL}" \
+  --channel "${LIBKVIKIO_CHANNEL}" \
   libcudf libcudf_kafka libcudf-tests libcudf-example
 
 rapids-logger "Check GPU usage"

diff --git a/ci/test_python_common.sh b/ci/test_python_common.sh
@@ -33,9 +33,11 @@ RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${RESULTS_DIR}/test-results"}/
 RAPIDS_COVERAGE_DIR=${RAPIDS_COVERAGE_DIR:-"${RESULTS_DIR}/coverage-results"}
 mkdir -p "${RAPIDS_TESTS_DIR}" "${RAPIDS_COVERAGE_DIR}"
 
+LIBKVIKIO_CHANNEL=$(rapids-get-pr-conda-artifact kvikio 479 cpp)
 rapids-print-env
 
 rapids-mamba-retry install \
   --channel "${CPP_CHANNEL}" \
   --channel "${PYTHON_CHANNEL}" \
+  --channel "${LIBKVIKIO_CHANNEL}" \
   cudf libcudf