Use mamba to build packages (#2051)

Using mamba should speed up CI Authors: - Jordan Jacobelli (https://github.com/Ethyling) Approvers: - AJ Schmidt (https://github.com/ajschmidt8) URL: #2051
rapidsai · May 23, 2022 · 1693a12 · 1693a12
1 parent 9b1d189
commit 1693a12
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 9 deletions.
diff --git a/ci/benchmark/build.sh b/ci/benchmark/build.sh
@@ -71,7 +71,7 @@ CUGRAPH_DEPS=(cudf rmm)
 LIBCUGRAPH_DEPS=(cudf rmm)
 
 gpuci_logger "Install required packages"
-gpuci_conda_retry install -c nvidia -c rapidsai -c rapidsai-nightly -c conda-forge \
+gpuci_mamba_retry install -c nvidia -c rapidsai -c rapidsai-nightly -c conda-forge \
  "cudf=${MINOR_VERSION}" \
  "rmm=${MINOR_VERSION}" \
  "cudatoolkit=$CUDA_REL" \

diff --git a/ci/cpu/build.sh b/ci/cpu/build.sh
@@ -78,6 +78,8 @@ else
  CONDA_LOCAL_CHANNEL="ci/artifacts/cugraph/cpu/.conda-bld/"
 fi
 
+# FIXME: Move boa install to gpuci/rapidsai
+gpuci_mamba_retry install -c conda-forge boa
 
 ###############################################################################
 # BUILD - Conda package builds
@@ -86,9 +88,9 @@ fi
 if [ "$BUILD_LIBCUGRAPH" == '1' ]; then
  gpuci_logger "Building conda package for libcugraph and libcugraph_etl"
  if [[ -z "$PROJECT_FLASH" || "$PROJECT_FLASH" == "0" ]]; then
- gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/libcugraph
+ gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/libcugraph
  else
- gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} --dirty --no-remove-work-dir conda/recipes/libcugraph
+ gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} --dirty --no-remove-work-dir conda/recipes/libcugraph
  mkdir -p ${CONDA_BLD_DIR}/libcugraph
  mv ${CONDA_BLD_DIR}/work ${CONDA_BLD_DIR}/libcugraph/work
  fi
@@ -101,11 +103,11 @@ fi
 if [ "$BUILD_CUGRAPH" == "1" ]; then
  gpuci_logger "Building conda packages for pylibcugraph and cugraph"
  if [[ -z "$PROJECT_FLASH" || "$PROJECT_FLASH" == "0" ]]; then
- gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/pylibcugraph --python=$PYTHON
- gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/cugraph --python=$PYTHON
+ gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/pylibcugraph --python=$PYTHON
+ gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/cugraph --python=$PYTHON
  else
- gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/pylibcugraph -c ${CONDA_LOCAL_CHANNEL} --dirty --no-remove-work-dir --python=$PYTHON
- gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/cugraph -c ${CONDA_LOCAL_CHANNEL} --dirty --no-remove-work-dir --python=$PYTHON
+ gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/pylibcugraph -c ${CONDA_LOCAL_CHANNEL} --dirty --no-remove-work-dir --python=$PYTHON
+ gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/cugraph -c ${CONDA_LOCAL_CHANNEL} --dirty --no-remove-work-dir --python=$PYTHON
  mkdir -p ${CONDA_BLD_DIR}/cugraph
  mv ${CONDA_BLD_DIR}/work ${CONDA_BLD_DIR}/cugraph/work
  fi

diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
@@ -106,11 +106,14 @@ else
  gpuci_logger "Installing libcugraph-tests"
  gpuci_mamba_retry install -c ${CONDA_ARTIFACT_PATH} libcugraph libcugraph_etl libcugraph-tests
 
+ # TODO: Move boa install to gpuci/rapidsai
+ gpuci_mamba_retry install boa
+
  gpuci_logger "Building and installing pylibcugraph and cugraph..."
  export CONDA_BLD_DIR="${WORKSPACE}/.conda-bld"
  export VERSION_SUFFIX=""
- gpuci_conda_retry build conda/recipes/pylibcugraph --no-build-id --croot ${CONDA_BLD_DIR} -c ${CONDA_ARTIFACT_PATH} --python=${PYTHON}
- gpuci_conda_retry build conda/recipes/cugraph --no-build-id --croot ${CONDA_BLD_DIR} -c ${CONDA_ARTIFACT_PATH} --python=${PYTHON}
+ gpuci_conda_retry mambabuild conda/recipes/pylibcugraph --no-build-id --croot ${CONDA_BLD_DIR} -c ${CONDA_ARTIFACT_PATH} --python=${PYTHON}
+ gpuci_conda_retry mambabuild conda/recipes/cugraph --no-build-id --croot ${CONDA_BLD_DIR} -c ${CONDA_ARTIFACT_PATH} --python=${PYTHON}
  gpuci_mamba_retry install cugraph pylibcugraph -c ${CONDA_BLD_DIR} -c ${CONDA_ARTIFACT_PATH}
 fi