Skip to content

Commit

Permalink
add a DefaultArrayStyle rule
Browse files Browse the repository at this point in the history
  • Loading branch information
pablosanjose committed Dec 10, 2017
1 parent 5d5bab6 commit b70fae3
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/broadcast.jl
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,11 @@ else

BroadcastStyle(::Type{<:StaticArray{D, T, N}}) where {D, T, N} = StaticArrayStyle{N}()

# Fix Precedence: Make StaticArray - Array -> Array
# Precedence rules: needed in particular to make StaticArray - Array -> Array
BroadcastStyle(::StaticArrayStyle{M}, ::Broadcast.VectorStyle) where M = Broadcast.Unknown()
BroadcastStyle(::StaticArrayStyle{M}, ::Broadcast.MatrixStyle) where M = Broadcast.Unknown()
BroadcastStyle(::StaticArrayStyle{M}, ::Broadcast.DefaultArrayStyle{N}) where {M,N} =
Broadcast.DefaultArrayStyle(Broadcast._max(Val(M), Val(N)))

# Add a broadcast method that calls the @generated routine
@inline function broadcast(f, ::StaticArrayStyle, ::Void, ::Void, As...)
Expand Down

0 comments on commit b70fae3

Please sign in to comment.