Skip to content

Commit

Permalink
Bump GPU CI to CUDA 11.8 (#1275)
Browse files Browse the repository at this point in the history
* Bump GPU CI to CUDA 11.8

* Resolve deprecation warnings
  • Loading branch information
charlesbluca authored Nov 28, 2023
1 parent 8269983 commit 779b6bb
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 14 deletions.
2 changes: 1 addition & 1 deletion continuous_integration/gpuci/axis.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ PYTHON_VER:
- "3.10"

CUDA_VER:
- "11.5.2"
- "11.8.0"

LINUX_VER:
- ubuntu20.04
Expand Down
26 changes: 13 additions & 13 deletions continuous_integration/gpuci/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,38 +27,38 @@ export CUDA_REL=${CUDA_VERSION%.*}
# SETUP - Check environment
################################################################################

gpuci_logger "Check environment variables"
rapids-logger "Check environment variables"
env

gpuci_logger "Check GPU usage"
rapids-logger "Check GPU usage"
nvidia-smi

gpuci_logger "Activate conda env"
rapids-logger "Activate conda env"
. /opt/conda/etc/profile.d/conda.sh
conda activate dask_sql

gpuci_logger "Update conda env"
rapids-logger "Update conda env"
gpuci_mamba_retry env update -n dask_sql -f continuous_integration/gpuci/environment-${PYTHON_VER}.yaml

gpuci_logger "Install awscli"
rapids-logger "Install awscli"
gpuci_mamba_retry install -y -c conda-forge awscli

gpuci_logger "Download parquet dataset"
gpuci_retry aws s3 cp --only-show-errors "${DASK_SQL_BUCKET_NAME}parquet_2gb_sorted/" tests/unit/data/ --recursive
rapids-logger "Download parquet dataset"
rapids-retry aws s3 cp --only-show-errors "${DASK_SQL_BUCKET_NAME}parquet_2gb_sorted/" tests/unit/data/ --recursive

gpuci_logger "Download query files"
gpuci_retry aws s3 cp --only-show-errors "${DASK_SQL_BUCKET_NAME}queries/" tests/unit/queries/ --recursive
rapids-logger "Download query files"
rapids-retry aws s3 cp --only-show-errors "${DASK_SQL_BUCKET_NAME}queries/" tests/unit/queries/ --recursive

gpuci_logger "Install dask-sql"
rapids-logger "Install dask-sql"
pip install -e . -vv

gpuci_logger "Check Python version"
rapids-logger "Check Python version"
python --version

gpuci_logger "Check conda environment"
rapids-logger "Check conda environment"
conda info
conda config --show-sources
conda list --show-channel-urls

gpuci_logger "Python py.test for dask-sql"
rapids-logger "Python py.test for dask-sql"
py.test $WORKSPACE -n 4 -v -m gpu --runqueries --rungpu --junitxml="$WORKSPACE/junit-dask-sql.xml" --cov-config="$WORKSPACE/.coveragerc" --cov=dask_sql --cov-report=xml:"$WORKSPACE/dask-sql-coverage.xml" --cov-report term

0 comments on commit 779b6bb

Please sign in to comment.