Add missing BroadcastStyle and MemoryLayouts for cache styles

Author: DanielVandH

Project.toml

@@ -1,6 +1,6 @@
 name = "LazyArrays"
 uuid = "5078a376-72f3-5289-bfd5-ec5146d43c02"
-version = "2.9"
+version = "2.9.1"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/cache.jl

@@ -333,6 +333,8 @@ MemoryLayout(C::Type{CachedArray{T,N,DAT,ARR}}) where {T,N,DAT,ARR} = cachedlayo
 
 MemoryLayout(::Type{<:AbstractCachedArray}) = GenericCachedLayout()
 
+transposelayout(::GenericCachedLayout) = GenericCachedLayout()
+
 #####
 # broadcasting
 #
@@ -345,10 +347,11 @@ CachedArrayStyle(::Val{N}) where N = CachedArrayStyle{N}()
 CachedArrayStyle{M}(::Val{N}) where {N,M} = CachedArrayStyle{N}()
 
 BroadcastStyle(::Type{<:AbstractCachedArray{<:Any,N}}) where N = CachedArrayStyle{N}()
+BroadcastStyle(::Type{<:AdjOrTrans{<:Any, <:AbstractCachedArray{<:Any,N}}}) where N = CachedArrayStyle{N}()
 BroadcastStyle(::Type{<:SubArray{<:Any,N,<:AbstractCachedArray{<:Any,M}}}) where {N,M} = CachedArrayStyle{M}()
+BroadcastStyle(::Type{<:SubArray{<:Any,N,<:AdjOrTrans{<:Any, <:AbstractCachedArray{<:Any,M}}}}) where {N,M} = CachedArrayStyle{M}()
 BroadcastStyle(::CachedArrayStyle{N}, ::LazyArrayStyle{M}) where {N,M} = CachedArrayStyle{max(M, N)}()
 
-
 broadcasted(::AbstractLazyArrayStyle, op, A::CachedArray) = CachedArray(broadcast(op, cacheddata(A)), broadcast(op, A.array))
 layout_broadcasted(::CachedLayout, _, op, A::AbstractArray, c::Number) = CachedArray(broadcast(op, cacheddata(A), c), broadcast(op, A.array, c))
 layout_broadcasted(_, ::CachedLayout, op, c::Number, A::CachedArray) = CachedArray(broadcast(op, c, cacheddata(A)), broadcast(op, c, A.array))

test/cachetests.jl

@@ -4,7 +4,7 @@ using LazyArrays, FillArrays, LinearAlgebra, ArrayLayouts, SparseArrays, Test
 using StaticArrays
 import LazyArrays: CachedArray, CachedMatrix, CachedVector, PaddedLayout, CachedLayout, resizedata!, zero!,
                     CachedAbstractArray, CachedAbstractVector, CachedAbstractMatrix, AbstractCachedArray, AbstractCachedMatrix,
-                    PaddedColumns, cacheddata, maybe_cacheddata
+                    PaddedColumns, cacheddata, maybe_cacheddata, Accumulate, CachedArrayStyle, GenericCachedLayout
 
 using ..InfiniteArrays
 using .InfiniteArrays: OneToInf
@@ -539,6 +539,18 @@ using Infinities
         C = [1, 2, 3]
         @test maybe_cacheddata(C) === C
     end
+
+    @testset "Missing BroadcastStyles/MemoryLayouts with CachedArrayStyles" begin
+        A = typeof(view(Accumulate(*, [1, 2, 3])', 1:1, 1:2))
+        B = typeof(view(transpose(Accumulate(*, [1, 2, 3])), 1:1, 1:2))
+        C = typeof(Accumulate(*, [1, 2, 3])')
+        @test Base.BroadcastStyle(A) == CachedArrayStyle{1}()
+        @test Base.BroadcastStyle(B) == CachedArrayStyle{1}()
+        @test Base.BroadcastStyle(C) == CachedArrayStyle{1}()
+        @test MemoryLayout(A) == GenericCachedLayout()
+        @test MemoryLayout(B) == GenericCachedLayout()
+        @test MemoryLayout(C) == DualLayout{GenericCachedLayout}()
+    end
 end
 
 end # module