diff --git a/ci/build_cpp.sh b/ci/build_cpp.sh index 0a6a649fd0..5cc41359b2 100755 --- a/ci/build_cpp.sh +++ b/ci/build_cpp.sh @@ -9,10 +9,16 @@ export CMAKE_GENERATOR=Ninja rapids-print-env +LIBRMM_CHANNEL=$(rapids-get-pr-conda-artifact rmm 1374 cpp) +LIBRAFT_CHANNEL=$(rapids-get-pr-conda-artifact raft 1957 cpp) + version=$(rapids-generate-version) rapids-logger "Begin cpp build" -RAPIDS_PACKAGE_VERSION=${version} rapids-conda-retry mambabuild conda/recipes/libcuml +RAPIDS_PACKAGE_VERSION=${version} rapids-conda-retry mambabuild \ + --channel "${LIBRMM_CHANNEL}" \ + --channel "${LIBRAFT_CHANNEL}" \ + conda/recipes/libcuml rapids-upload-conda-to-s3 cpp diff --git a/ci/build_python.sh b/ci/build_python.sh index 1332062770..9c6d3c8793 100755 --- a/ci/build_python.sh +++ b/ci/build_python.sh @@ -14,7 +14,7 @@ package_dir="python" version=$(rapids-generate-version) git_commit=$(git rev-parse HEAD) -export RAPIDS_PACKAGE_VERSION=${version} +export RAPIDS_PACKAGE_VERSION=${version} echo "${version}" > VERSION sed -i "/^__git_commit__/ s/= .*/= \"${git_commit}\"/g" "${package_dir}/${package_name}/_version.py" @@ -22,12 +22,24 @@ sed -i "/^__git_commit__/ s/= .*/= \"${git_commit}\"/g" "${package_dir}/${packag rapids-logger "Begin py build" CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp) +LIBRMM_CHANNEL=$(rapids-get-pr-conda-artifact rmm 1374 cpp) +RMM_CHANNEL=$(rapids-get-pr-conda-artifact rmm 1374 python) +LIBRAFT_CHANNEL=$(rapids-get-pr-conda-artifact raft 1957 cpp) +RAFT_CHANNEL=$(rapids-get-pr-conda-artifact raft 1957 python) +LIBCUDF_CHANNEL=$(rapids-get-pr-conda-artifact cudf 14355 cpp) +CUDF_CHANNEL=$(rapids-get-pr-conda-artifact cudf 14355 python) # TODO: Remove `--no-test` flag once importing on a CPU # node works correctly rapids-conda-retry mambabuild \ --no-test \ --channel "${CPP_CHANNEL}" \ + --channel "${LIBRMM_CHANNEL}" \ + --channel "${RMM_CHANNEL}" \ + --channel "${LIBRAFT_CHANNEL}" \ + --channel "${RAFT_CHANNEL}" \ + --channel "${LIBCUDF_CHANNEL}" \ + --channel "${CUDF_CHANNEL}" \ conda/recipes/cuml # Build cuml-cpu only in CUDA 11 jobs since it only depends on python diff --git a/ci/test_cpp.sh b/ci/test_cpp.sh index ebed9c9bdc..593962f5d7 100755 --- a/ci/test_cpp.sh +++ b/ci/test_cpp.sh @@ -19,6 +19,8 @@ conda activate test set -u CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp) +LIBRMM_CHANNEL=$(rapids-get-pr-conda-artifact rmm 1374 cpp) +LIBRAFT_CHANNEL=$(rapids-get-pr-conda-artifact raft 1957 cpp) RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"}/ mkdir -p "${RAPIDS_TESTS_DIR}" @@ -26,6 +28,8 @@ rapids-print-env rapids-mamba-retry install \ --channel "${CPP_CHANNEL}" \ + --channel "${LIBRMM_CHANNEL}" \ + --channel "${LIBRAFT_CHANNEL}" \ libcuml libcuml-tests rapids-logger "Check GPU usage" diff --git a/ci/test_python_common.sh b/ci/test_python_common.sh index 1ec5d59380..9cedbf12a0 100644 --- a/ci/test_python_common.sh +++ b/ci/test_python_common.sh @@ -21,6 +21,12 @@ set -u rapids-logger "Downloading artifacts from previous jobs" CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp) PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python) +LIBRMM_CHANNEL=$(rapids-get-pr-conda-artifact rmm 1374 cpp) +RMM_CHANNEL=$(rapids-get-pr-conda-artifact rmm 1374 python) +LIBRAFT_CHANNEL=$(rapids-get-pr-conda-artifact raft 1957 cpp) +RAFT_CHANNEL=$(rapids-get-pr-conda-artifact raft 1957 python) +LIBCUDF_CHANNEL=$(rapids-get-pr-conda-artifact cudf 14355 cpp) +CUDF_CHANNEL=$(rapids-get-pr-conda-artifact cudf 14355 python) RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"} RAPIDS_COVERAGE_DIR=${RAPIDS_COVERAGE_DIR:-"${PWD}/coverage-results"} @@ -31,6 +37,12 @@ rapids-print-env rapids-mamba-retry install \ --channel "${CPP_CHANNEL}" \ --channel "${PYTHON_CHANNEL}" \ + --channel "${LIBRMM_CHANNEL}" \ + --channel "${RMM_CHANNEL}" \ + --channel "${LIBRAFT_CHANNEL}" \ + --channel "${RAFT_CHANNEL}" \ + --channel "${LIBCUDF_CHANNEL}" \ + --channel "${CUDF_CHANNEL}" \ libcuml cuml rapids-logger "Check GPU usage" diff --git a/fetch_rapids.cmake b/fetch_rapids.cmake index 98ce6888ba..de92797dc2 100644 --- a/fetch_rapids.cmake +++ b/fetch_rapids.cmake @@ -11,6 +11,8 @@ # or implied. See the License for the specific language governing permissions and limitations under # the License. # ============================================================================= +set(rapids-cmake-repo kkraus14/rapids-cmake) +set(rapids-cmake-branch spdlog_1.12_fmt_10) if(NOT EXISTS ${CMAKE_CURRENT_BINARY_DIR}/CUML_RAPIDS.cmake) file(DOWNLOAD https://raw.githubusercontent.com/rapidsai/rapids-cmake/branch-24.02/RAPIDS.cmake ${CMAKE_CURRENT_BINARY_DIR}/CUML_RAPIDS.cmake