From b3fb2c7675397ac233c1be4abfeaa3e5a3d3d76c Mon Sep 17 00:00:00 2001 From: Sheehan Olver Date: Tue, 7 Nov 2023 20:18:21 +0000 Subject: [PATCH] Support Transpose broadcasting --- src/blockbanded/blockbanded.jl | 1 + 1 file changed, 1 insertion(+) diff --git a/src/blockbanded/blockbanded.jl b/src/blockbanded/blockbanded.jl index 59ae560..132208b 100644 --- a/src/blockbanded/blockbanded.jl +++ b/src/blockbanded/blockbanded.jl @@ -64,6 +64,7 @@ BroadcastStyle(::Type{<:SubArray{T,N,Arr,<:NTuple{N,BlockSlice{BlockRange{1,Tupl # TODO: generalise following BroadcastStyle(::Type{<:BlockArray{T,N,<:AbstractArray{<:AbstractArray{T,N},N},<:NTuple{N,BlockedUnitRange{<:InfRanges}}}}) where {T,N} = LazyArrayStyle{N}() +BroadcastStyle(::Type{<:Transpose{T,<:BlockArray{T,N,<:AbstractArray{<:AbstractArray{T,N},N},<:NTuple{N,BlockedUnitRange{<:InfRanges}}}}}) where {T,N} = LazyArrayStyle{2}() # BroadcastStyle(::Type{<:PseudoBlockArray{T,N,<:AbstractArray{T,N},<:NTuple{N,BlockedUnitRange{<:InfRanges}}}}) where {T,N} = LazyArrayStyle{N}() BroadcastStyle(::Type{<:BlockArray{T,N,<:AbstractArray{<:AbstractArray{T,N},N},<:NTuple{N,BlockedUnitRange{<:RangeCumsum{Int,<:InfRanges}}}}}) where {T,N} = LazyArrayStyle{N}() # BroadcastStyle(::Type{<:PseudoBlockArray{T,N,<:AbstractArray{T,N},<:NTuple{N,BlockedUnitRange{<:RangeCumsum{Int,<:InfRanges}}}}}) where {T,N} = LazyArrayStyle{N}()