diff --git a/Project.toml b/Project.toml index e043be498f..50504b7c13 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "ClimaCore" uuid = "d414da3d-4745-48bb-8d80-42e94e092884" authors = ["CliMA Contributors "] -version = "0.14.2" +version = "0.14.3" [deps] Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e" diff --git a/ext/cuda/data_layouts.jl b/ext/cuda/data_layouts.jl index 7eec9dfe21..ccbbff4ba3 100644 --- a/ext/cuda/data_layouts.jl +++ b/ext/cuda/data_layouts.jl @@ -11,6 +11,8 @@ import Adapt import CUDA device_from_array_type(::Type{<:CUDA.CuArray}) = ClimaComms.CUDADevice() +device_from_array_type(::Type{<:SubArray{<:Any, <:Any, <:CUDA.CuArray}}) = + ClimaComms.CUDADevice() parent_array_type(::Type{<:CUDA.CuArray{T, N, B} where {N}}) where {T, B} = CUDA.CuArray{T, N, B} where {N}