From 3f4dbd27f29e4da6e4840e4526d64112e4dbf464 Mon Sep 17 00:00:00 2001
From: Charles Blackmon-Luca <20627856+charlesbluca@users.noreply.github.com>
Date: Tue, 21 May 2024 02:09:44 -0400
Subject: [PATCH] Add python 3.11 build to GPU CI (#376)

* Add python 3.11 build to GPU CI

* Switch to `cupy-cuda11x`

Previously CuPy built a wheel for each CUDA `X.Y` version. However the
CuPy team later transitioned to CUDA Compatibility, which allows a wheel
build to only focus on CUDA major version. So switch to that wheel here.

---------

Co-authored-by: jakirkham <jakirkham@gmail.com>
---
 continuous_integration/gpuci/axis.yaml | 1 +
 continuous_integration/gpuci/build.sh  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/continuous_integration/gpuci/axis.yaml b/continuous_integration/gpuci/axis.yaml
index 30ce0c98..05e9edaf 100644
--- a/continuous_integration/gpuci/axis.yaml
+++ b/continuous_integration/gpuci/axis.yaml
@@ -1,6 +1,7 @@
 PYTHON_VER:
 - "3.9"
 - "3.10"
+- "3.11"
 
 CUDA_VER:
 - "11.8.0"
diff --git a/continuous_integration/gpuci/build.sh b/continuous_integration/gpuci/build.sh
index cf887f90..9b811b2f 100644
--- a/continuous_integration/gpuci/build.sh
+++ b/continuous_integration/gpuci/build.sh
@@ -41,7 +41,7 @@ rapids-logger "Activate conda env"
 conda activate dask_image
 
 rapids-logger "Install cupy"
-python -m pip install cupy-cuda112 -f https://pip.cupy.dev/pre
+python -m pip install cupy-cuda11x -f https://pip.cupy.dev/pre
 
 rapids-logger "Install dask-image"
 python -m pip install .