From c79ea1eb3d9b1f7155be5593b5fa48389883b163 Mon Sep 17 00:00:00 2001
From: Jordan Jacobelli <jjacobelli@nvidia.com>
Date: Wed, 27 Oct 2021 11:13:09 +0200
Subject: [PATCH] Build conda packages using mambabuild

Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com>
---
 ci/cpu/build.sh | 11 +++++++----
 ci/gpu/build.sh |  4 +++-
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/ci/cpu/build.sh b/ci/cpu/build.sh
index c042db739..7b1208758 100755
--- a/ci/cpu/build.sh
+++ b/ci/cpu/build.sh
@@ -55,6 +55,9 @@ conda list --show-channel-urls
 # FIX Added to deal with Anancoda SSL verification issues during conda builds
 conda config --set ssl_verify False
 
+# FIXME: Move installation to gpuci/rapidsai images
+gpuci_mamba_retry install -c conda-forge boa
+
 ################################################################################
 # BUILD - Conda package builds (conda deps: librmm <- rmm)
 ################################################################################
@@ -62,9 +65,9 @@ conda config --set ssl_verify False
 if [[ "$BUILD_LIBRMM" == "1" ]]; then
   gpuci_logger "Build conda pkg for librmm"
   if [[ -z "$PROJECT_FLASH" || "$PROJECT_FLASH" == "0" ]]; then
-    gpuci_conda_retry build conda/recipes/librmm --python=$PYTHON
+    gpuci_conda_retry mambabuild conda/recipes/librmm --python=$PYTHON
   else
-    gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} --dirty --no-remove-work-dir conda/recipes/librmm
+    gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} --dirty --no-remove-work-dir conda/recipes/librmm
     mkdir -p ${CONDA_BLD_DIR}/librmm
     mv ${CONDA_BLD_DIR}/work/ ${CONDA_BLD_DIR}/librmm/work
   fi
@@ -75,9 +78,9 @@ fi
 if [[ "$BUILD_RMM" == "1" ]]; then
   gpuci_logger "Build conda pkg for rmm"
   if [[ -z "$PROJECT_FLASH" || "$PROJECT_FLASH" == "0" ]]; then
-    gpuci_conda_retry build conda/recipes/rmm --python=$PYTHON
+    gpuci_conda_retry mambabuild conda/recipes/rmm --python=$PYTHON
   else
-    gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} --dirty --no-remove-work-dir \
+    gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} --dirty --no-remove-work-dir \
       -c $WORKSPACE/ci/artifacts/rmm/cpu/.conda-bld/ conda/recipes/rmm
     mkdir -p ${CONDA_BLD_DIR}/rmm
     mv ${CONDA_BLD_DIR}/work/ ${CONDA_BLD_DIR}/rmm/work
diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
index 9176d4b15..2689667fb 100755
--- a/ci/gpu/build.sh
+++ b/ci/gpu/build.sh
@@ -111,9 +111,11 @@ else
         fi
     done
 
+    # TODO: Move boa install to gpuci/rapidsai
+    gpuci_mamba_retry install boa
     export CONDA_BLD_DIR="$WORKSPACE/.conda-bld"
     gpuci_logger "Building and installing rmm"
-    gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} \
+    gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} \
       -c $WORKSPACE/ci/artifacts/rmm/cpu/.conda-bld/ conda/recipes/rmm
     gpuci_mamba_retry install -c $WORKSPACE/ci/artifacts/rmm/cpu/.conda-bld/ \
       -c ${CONDA_BLD_DIR} rmm