Build conda packages using mambabuild

Signed-off-by: Jordan Jacobelli <[email protected]>
rapidsai · Oct 27, 2021 · 2602404 · 2602404
1 parent 0fbe357
commit 2602404
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/ci/cpu/build.sh b/ci/cpu/build.sh
@@ -58,21 +58,24 @@ conda config --set ssl_verify False
 # BUILD - Conda package builds (conda deps: librmm <- rmm)
 ################################################################################
 
+# TODO: Remove this
+gpuci_conda_retry install -c conda-forge boa
+
 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 --dirty --no-remove-work-dir conda/recipes/librmm
+    gpuci_conda_retry mambabuild --dirty --no-remove-work-dir conda/recipes/librmm
   fi
 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 --dirty --no-remove-work-dir \
+    gpuci_conda_retry mambabuild --dirty --no-remove-work-dir \
       -c $WORKSPACE/ci/artifacts/rmm/cpu/conda-bld/ conda/recipes/rmm
 
   fi

diff --git a/ci/cpu/upload.sh b/ci/cpu/upload.sh
@@ -11,6 +11,9 @@ export GPUCI_RETRY_SLEEP=30
 # Set default label options if they are not defined elsewhere
 export LABEL_OPTION=${LABEL_OPTION:-"--label main"}
 
+# TODO: Remove
+conda mambabuild conda/recipes/librmm --output
+
 # Skip uploads unless BUILD_MODE == "branch"
 if [ ${BUILD_MODE} != "branch" ]; then
   echo "Skipping upload"
@@ -29,8 +32,8 @@ fi
 
 gpuci_logger "Get conda file output locations"
 
-export LIBRMM_FILE=`conda build conda/recipes/librmm --output`
-export RMM_FILE=`conda build conda/recipes/rmm --python=$PYTHON --output`
+export LIBRMM_FILE=`conda mambabuild conda/recipes/librmm --output`
+export RMM_FILE=`conda mambabuild conda/recipes/rmm --python=$PYTHON --output`
 
 ################################################################################
 # UPLOAD - Conda packages