From 779b6bb05cae18044f0676acb383a6e2201ec920 Mon Sep 17 00:00:00 2001
From: Charles Blackmon-Luca <20627856+charlesbluca@users.noreply.github.com>
Date: Tue, 28 Nov 2023 14:43:04 -0500
Subject: [PATCH] Bump GPU CI to CUDA 11.8 (#1275)

* Bump GPU CI to CUDA 11.8

* Resolve deprecation warnings
---
 continuous_integration/gpuci/axis.yaml |  2 +-
 continuous_integration/gpuci/build.sh  | 26 +++++++++++++-------------
 2 files changed, 14 insertions(+), 14 deletions(-)

diff --git a/continuous_integration/gpuci/axis.yaml b/continuous_integration/gpuci/axis.yaml
index dd7525507..83a47500c 100644
--- a/continuous_integration/gpuci/axis.yaml
+++ b/continuous_integration/gpuci/axis.yaml
@@ -3,7 +3,7 @@ PYTHON_VER:
 - "3.10"
 
 CUDA_VER:
-- "11.5.2"
+- "11.8.0"
 
 LINUX_VER:
 - ubuntu20.04
diff --git a/continuous_integration/gpuci/build.sh b/continuous_integration/gpuci/build.sh
index a26d236b9..156a945f0 100644
--- a/continuous_integration/gpuci/build.sh
+++ b/continuous_integration/gpuci/build.sh
@@ -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