From 03da55575843c507e8f8693a6ab1927865683b75 Mon Sep 17 00:00:00 2001
From: Charles Blackmon-Luca <20627856+charlesbluca@users.noreply.github.com>
Date: Mon, 20 May 2024 13:45:49 -0700
Subject: [PATCH 1/2] Add python 3.11 build to GPU CI

---
 continuous_integration/gpuci/axis.yaml | 1 +
 1 file changed, 1 insertion(+)

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"

From a229fe0d81364d3e5ef7e9d372f95c0f406e87f6 Mon Sep 17 00:00:00 2001
From: jakirkham <jakirkham@gmail.com>
Date: Mon, 20 May 2024 14:08:45 -0700
Subject: [PATCH 2/2] 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.
---
 continuous_integration/gpuci/build.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 .