From 5c0b7417e8dee1ae53784034fe053875cc7632a7 Mon Sep 17 00:00:00 2001 From: Charles Kawczynski Date: Thu, 9 May 2024 22:48:28 -0400 Subject: [PATCH 1/2] Fix MultiBroadcastFusion dispatch --- ext/cuda/data_layouts.jl | 2 ++ 1 file changed, 2 insertions(+) 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} From d543a226ed597ae5185a8908ef1d06d47d700c00 Mon Sep 17 00:00:00 2001 From: Charles Kawczynski Date: Thu, 9 May 2024 22:50:51 -0400 Subject: [PATCH 2/2] Bump patch --- Project.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"