Skip to content

Commit

Permalink
Use conda to build python packages during GPU tests
Browse files Browse the repository at this point in the history
Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com>
  • Loading branch information
jjacobelli committed Apr 20, 2022
1 parent c4700e6 commit 9bcb67c
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 5 deletions.
15 changes: 10 additions & 5 deletions ci/gpu/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ export HOME="$WORKSPACE"
cd "$WORKSPACE"
export GIT_DESCRIBE_TAG=`git describe --tags`
export MINOR_VERSION=`echo $GIT_DESCRIBE_TAG | grep -o -E '([0-9]+\.[0-9]+)'`
unset GIT_DESCRIBE_TAG

################################################################################
# SETUP - Check environment
Expand Down Expand Up @@ -66,6 +67,7 @@ gpuci_logger "Clone cudf"
git clone https://github.com/rapidsai/cudf.git -b branch-$MINOR_VERSION ${CUDF_HOME}
cd $CUDF_HOME
git submodule update --init --remote --recursive
cd "${WORKSPACE}"

if [[ -z "$PROJECT_FLASH" || "$PROJECT_FLASH" == "0" ]]; then
################################################################################
Expand Down Expand Up @@ -113,8 +115,15 @@ else
gpuci_logger "Check GPU usage"
nvidia-smi

gpuci_logger "Installing libcuspatial and libcuspatial-tests"
gpuci_mamba_retry install -c "${CONDA_ARTIFACT_PATH}" libcuspatial libcuspatial-tests

gpuci_logger "Building and installing cuspatial"
export CONDA_BLD_DIR="${WORKSPACE}/.conda-bld"
export VERSION_SUFFIX=""
gpuci_conda_retry build --croot "${CONDA_BLD_DIR}" -c "${CONDA_ARTIFACT_PATH}" -c "${CONDA_BLD_DIR}" conda/recipes/cuspatial
gpuci_mamba_retry install -c "${CONDA_BLD_DIR}" -c "${CONDA_ARTIFACT_PATH}" cuspatial

gpuci_logger "Running googletests"
for gt in "${CONDA_PREFIX}/bin/gtests/libcuspatial/"*; do
test_name=$(basename ${gt})
Expand All @@ -127,12 +136,8 @@ else
fi
done

cd "$WORKSPACE/python"

gpuci_logger "Building cuspatial"
"$WORKSPACE/build.sh" -v cuspatial

gpuci_logger "Run pytests"
cd "$WORKSPACE/python/cuspatial/cuspatial"
py.test --cache-clear --junitxml="$WORKSPACE/junit-cuspatial.xml" -v

EXITCODE=$?
Expand Down
Empty file.

0 comments on commit 9bcb67c

Please sign in to comment.