CompatHelper: bump compat for GPUArrays to 11 and JLArrays to 0.2, (keep existing compat)

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## Unreleased
 
+- update to GPUArrays v11, JLArrays v0.2 and remove CUDA from tests [#590](https://github.com/SpeedyWeather/SpeedyWeather.jl/pull/590)
 - Power spectrum for n-dim LowerTriangularArrays [#618](https://github.com/SpeedyWeather/SpeedyWeather.jl/pull/618)
 - Added custom EnzymeRules for the SpectralTransform and an extension for compatibility with FiniteDifferences.jl [#589](https://github.com/SpeedyWeather/SpeedyWeather.jl/pull/589)
 - Optical depth introduced and array-agnostic ColumnVariables [#606](https://github.com/SpeedyWeather/SpeedyWeather.jl/pull/606)

Project.toml

@@ -54,10 +54,10 @@ Dates = "1.10"
 DocStringExtensions = "0.9"
 FFTW = "1"
 FastGaussQuadrature = "0.4, 0.5, 1"
-GPUArrays = "10"
+GPUArrays = "11"
 GenericFFT = "0.1"
-JLArrays = "0.1"
 GeoMakie = "0.7.6"
+JLArrays = "0.1.4, 0.2"
 JLD2 = "0.4, 0.5"
 KernelAbstractions = "0.9"
 LinearAlgebra = "1.10"

docs/Project.toml

@@ -6,8 +6,8 @@ NCDatasets = "85f8d34a-cbdd-5861-8df4-14fed0d494ab"
 UnicodePlots = "b8865327-cd53-5732-bb35-84acbb429228"
 
 [compat]
-CairoMakie = "0.11, 0.12"
+CairoMakie = "0.12"
 Documenter = "0.26, 0.27, 1"
-GeoMakie = "0.7.5"
+GeoMakie = "0.7.6"
 NCDatasets = "0.12, 0.13, 0.14"
 UnicodePlots = "3"

src/LowerTriangularMatrices/LowerTriangularMatrices.jl

@@ -6,6 +6,7 @@ using DocStringExtensions
 # GPU
 import Adapt
 import GPUArrays
+import KernelAbstractions
 
 # NUMERICS
 import LinearAlgebra: tril!

src/LowerTriangularMatrices/lower_triangular_array.jl

@@ -630,10 +630,10 @@ function Base.similar(
     return LowerTriangularArray{T, N, ArrayType{T,N}}(undef, size(L; as=Matrix))
 end
 
-function GPUArrays.backend(
-    ::Type{LowerTriangularArray{T, N, ArrayType}}
+function KernelAbstractions.get_backend(
+    a::LowerTriangularArray{T, N, ArrayType}
 ) where {T, N, ArrayType <: GPUArrays.AbstractGPUArray}
-    return GPUArrays.backend(ArrayType)
+    return KernelAbstractions.get_backend(a.data)
 end
 
 Adapt.adapt_structure(to, L::LowerTriangularArray) =

src/RingGrids/RingGrids.jl

@@ -12,6 +12,7 @@ import LinearAlgebra
 # GPU
 import Adapt
 import GPUArrays
+import KernelAbstractions
 
 # ABSTRACT GRIDS (2D) AND GRIDARRAYS (3D+)
 export  AbstractGridArray,

src/RingGrids/general.jl

@@ -482,10 +482,10 @@ AbstractGPUGridArrayStyle{2, ArrayType, Grid}(::Val{1}) where {ArrayType, Grid}
 AbstractGPUGridArrayStyle{3, ArrayType, Grid}(::Val{4}) where {ArrayType, Grid} = AbstractGPUGridArrayStyle{4, ArrayType, Grid}()
 AbstractGPUGridArrayStyle{3, ArrayType, Grid}(::Val{2}) where {ArrayType, Grid} = AbstractGPUGridArrayStyle{3, ArrayType, Grid}()
 
-function GPUArrays.backend(
-    ::Type{Grid}
+function KernelAbstractions.get_backend(
+    g::Grid
 ) where {Grid <: AbstractGridArray{T, N, ArrayType}} where {T, N, ArrayType <: GPUArrays.AbstractGPUArray}
-    return GPUArrays.backend(ArrayType)
+    return KernelAbstractions.get_backend(g.data)
 end
 
 function Base.similar(

test/Project.toml

@@ -1,7 +1,6 @@
 [deps]
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
-CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
 Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
 EnzymeTestUtils = "12d8515a-0907-448a-8884-5fe00fdf1c5a"