Update cupy dependency (#13284)

Switch to requiring cupy 12

Authors:
  - Vyas Ramasubramani (https://github.com/vyasr)

Approvers:
  - Jake Awe (https://github.com/AyodeAwe)
  - Ashwin Srinath (https://github.com/shwina)

URL: #13284

.github/workflows/pr.yaml

@@ -112,7 +112,7 @@ jobs:
       build_type: pull-request
       package-name: cudf
       # Install cupy-cuda11x for arm from a special index url
-      test-before-arm64: "python -m pip install 'cupy-cuda11x<12.0.0' -f https://pip.cupy.dev/aarch64"
+      test-before-arm64: "python -m pip install 'cupy-cuda11x>=12.0.0' -f https://pip.cupy.dev/aarch64"
       test-unittest: "python -m pytest -n 8 ./python/cudf/cudf/tests"
       test-smoketest: "python ./ci/wheel_smoke_test_cudf.py"
   wheel-build-dask-cudf:

.github/workflows/test.yaml

@@ -86,7 +86,7 @@ jobs:
       date: ${{ inputs.date }}
       sha: ${{ inputs.sha }}
       package-name: cudf
-      test-before-arm64: "python -m pip install 'cupy-cuda11x<12.0.0' -f https://pip.cupy.dev/aarch64"
+      test-before-arm64: "python -m pip install 'cupy-cuda11x>=12.0.0' -f https://pip.cupy.dev/aarch64"
       test-unittest: "python -m pytest -n 8 ./python/cudf/cudf/tests"
   wheel-tests-dask-cudf:
     secrets: inherit

conda/environments/all_cuda-118_arch-x86_64.yaml

@@ -18,7 +18,7 @@ dependencies:
 - cuda-python>=11.7.1,<12.0
 - cuda-sanitizer-api=11.8.86
 - cudatoolkit=11.8
-- cupy>=9.5.0,<12.0.0a0
+- cupy>=12.0.0
 - cxx-compiler
 - cython>=0.29,<0.30
 - dask-core==2023.3.2

conda/recipes/cudf/meta.yaml

@@ -64,7 +64,7 @@ requirements:
     - python
     - typing_extensions
     - pandas >=1.3,<1.6.0dev0
-    - cupy >=9.5.0,<12.0.0a0
+    - cupy >=12.0.0
     - numba >=0.56.4,<0.57
     - numpy >=1.21
     - {{ pin_compatible('pyarrow', max_pin='x.x.x') }}

dependencies.yaml

@@ -358,7 +358,7 @@ dependencies:
       - output_types: conda
         packages:
           - cubinlinker
-          - cupy>=9.5.0,<12.0.0a0
+          - cupy>=12.0.0
           - pip
           - pip:
               - git+https://github.com/python-streamz/streamz.git@master
@@ -374,15 +374,15 @@ dependencies:
       - output_types: pyproject
         packages:
           - cubinlinker
-          - &cupy_pip cupy-cuda11x>=9.5.0,<12.0.0a0
+          - &cupy_pip cupy-cuda11x>=12.0.0
           - ptxcompiler
     specific:
       - output_types: requirements
         matrices:
           - matrix:
               arch: x86_64
             packages:
-              - cupy-cuda115>=9.5.0,<12.0.0a0 # TODO: This might change to cupy-cuda11x?
+              - cupy-cuda115>=12.0.0
           - matrix:
               arch: aarch64
             packages:

python/cudf/cudf/tests/test_array_ufunc.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2020-2022, NVIDIA CORPORATION.
+# Copyright (c) 2020-2023, NVIDIA CORPORATION.
 
 import operator
 import warnings
@@ -358,6 +358,7 @@ def test_ufunc_dataframe(request, ufunc, has_nulls, indexed):
                     "fmax",
                     "fmin",
                     "fmod",
+                    "heaviside",
                     "gcd",
                     "hypot",
                     "lcm",

python/cudf/pyproject.toml

@@ -29,7 +29,7 @@ dependencies = [
     "cachetools",
     "cubinlinker",
     "cuda-python>=11.7.1,<12.0",
-    "cupy-cuda11x>=9.5.0,<12.0.0a0",
+    "cupy-cuda11x>=12.0.0",
     "fsspec>=0.6.0",
     "numba>=0.56.4,<0.57",
     "numpy>=1.21",

python/dask_cudf/pyproject.toml

@@ -19,7 +19,7 @@ license = { text = "Apache 2.0" }
 requires-python = ">=3.9"
 dependencies = [
     "cudf==23.6.*",
-    "cupy-cuda11x>=9.5.0,<12.0.0a0",
+    "cupy-cuda11x>=12.0.0",
     "dask==2023.3.2",
     "distributed==2023.3.2.1",
     "fsspec>=0.6.0",