Update to CUDNN 9. (#2267)

[skip julia]
[skip cuda]

Author: maleadt

.buildkite/pipeline.yml

@@ -149,7 +149,7 @@ steps:
   - group: ":telescope: Downstream"
     depends_on: "cuda"
     steps:
-      - label: "NNlibCUDA.jl on CUDA 12"
+      - label: "NNlib.jl on CUDA 12"
         plugins:
           - JuliaCI/julia#v1:
               version: "1.10"
@@ -167,9 +167,9 @@ steps:
               Pkg.instantiate()
 
               try
-                Pkg.develop("NNlibCUDA")
+                Pkg.develop("NNlib")
               catch err
-                @error "Could not install NNlibCUDA" exception=(err,catch_backtrace())
+                @error "Could not install NNlib" exception=(err,catch_backtrace())
                 exit(3)
               end
               Pkg.develop(path=joinpath(pwd(), "lib", "cudnn"))
@@ -178,7 +178,10 @@ steps:
             end
 
             println("+++ :julia: Running tests")
-            Pkg.test("NNlibCUDA"; coverage=true)'
+            Pkg.test("NNlib"; coverage=true)'
+        env:
+          NNLIB_TEST_CUDA: "true"
+          NNLIB_TEST_CPU: "false"
         agents:
           queue: "juliagpu"
           cuda: "*"

lib/cudnn/Project.toml

@@ -13,5 +13,5 @@ CUDNN_jll = "62b44479-cb7b-5706-934f-f13b2eb2e645"
 CEnum = "0.2, 0.3, 0.4, 0.5"
 CUDA = "~5.3"
 CUDA_Runtime_Discovery = "0.2"
-CUDNN_jll = "~8.9"
+CUDNN_jll = "~9.0"
 julia = "1.8"