Skip to content

Commit e787db6

Browse files
committed
fix: Add two more methods of BroadcastStyle for disambiguation
1 parent ad4b580 commit e787db6

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/fft.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,8 +166,10 @@ Broadcast.BroadcastStyle(T::Type{<:FFTOut}) = ArrayStyle{T}()
166166

167167
# TODO: Consider instead using a `promote_rule` definition to promote `RFFTOut` to `FFTOut` <02-09-25>
168168
Broadcast.BroadcastStyle(a::ArrayStyle{A}, ::ArrayStyle{B}) where {T1,T2,M,N,A<:FFTOut{T1, M}, B<:RFFTOut{T2,N}} = ArrayStyle{FFTOut{promote_type(T1, T2), max(N,M)}}()
169+
Broadcast.BroadcastStyle(a::ArrayStyle{A}, ::ArrayStyle{B}) where {T1,T2,M,N,A<:FFTOut{T1, M}, B<:FFTOut{T2,N}} = ArrayStyle{FFTOut{promote_type(T1, T2), max(N,M)}}()
169170
Broadcast.BroadcastStyle(a::ArrayStyle{<:FFTOut{T, M}}, ::DefaultArrayStyle{N}) where {T, M, N} = ArrayStyle{FFTOut{T, max(N,M)}}()
170171
Broadcast.BroadcastStyle(a::ArrayStyle{<:FFTOut{T, M}}, ::AbstractArrayStyle{N}) where {T, M, N} = ArrayStyle{FFTOut{T, max(N,M)}}()
172+
Broadcast.BroadcastStyle(::ArrayStyle{A}, ::ArrayStyle{B}) where {B<:FFTOut, A<:FFTOut} = ArrayStyle{promote_type(A,B)}()
171173

172174
Base.similar(bc::Broadcasted{<:ArrayStyle{<:FFTOut}}, ::Type{S}) where {S} = FFTOut(similar(Array{S}, axes(bc)))
173175

0 commit comments

Comments
 (0)