Skip to content

Commit

Permalink
fix project flash
Browse files Browse the repository at this point in the history
  • Loading branch information
ajschmidt8 committed Jan 28, 2022
1 parent 26df825 commit a6a0509
Showing 1 changed file with 9 additions and 5 deletions.
14 changes: 9 additions & 5 deletions ci/cpu/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ set -e
export PATH=/opt/conda/bin:/usr/local/cuda/bin:$PATH
export PARALLEL_LEVEL=${PARALLEL_LEVEL:-4}
export CMAKE_GENERATOR="Ninja"
export CONDA_BLD_DIR="$WORKSPACE/conda-bld"

# Set home to the job's workspace
export HOME=$WORKSPACE
Expand Down Expand Up @@ -66,20 +67,23 @@ 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 --no-build-id conda/recipes/librmm --python=$PYTHON
gpuci_conda_retry build conda/recipes/librmm --python=$PYTHON
else
gpuci_conda_retry build --no-build-id --dirty --no-remove-work-dir conda/recipes/librmm
gpuci_conda_retry build --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
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 --no-build-id conda/recipes/rmm --python=$PYTHON
gpuci_conda_retry build conda/recipes/rmm --python=$PYTHON
else
gpuci_conda_retry build --no-build-id --dirty --no-remove-work-dir \
gpuci_conda_retry build --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
fi
fi

Expand Down

0 comments on commit a6a0509

Please sign in to comment.