Skip to content

Commit 6652793

Browse files
committed
Fix broadcast issue
1 parent 3be9661 commit 6652793

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/broadcast.jl

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,11 @@ end
1212
_broadcast(f, (Size(), Size(x)), T, x)
1313
end
1414

15-
@inline broadcast_sizes(a...) = _broadcast_sizes((), a...)
16-
@inline _broadcast_sizes(t::Tuple) = t
17-
@inline _broadcast_sizes(t::Tuple, a::StaticArray, as...) = _broadcast_sizes((t..., Size(a)), as...)
18-
@inline _broadcast_sizes(t::Tuple, a::Number, as...) = _broadcast_sizes((t..., Size()), as...)
15+
@inline broadcast_sizes(a...) = _broadcast_sizes(a...)
16+
@inline _broadcast_sizes(a::StaticArray) = (Size(a),)
17+
@inline _broadcast_sizes(a::Number) = (Size(),)
18+
@inline _broadcast_sizes(a::StaticArray, as...) = (Size(a), _broadcast_sizes(as...)...)
19+
@inline _broadcast_sizes(a::Number, as...) = (Size(), _broadcast_sizes(as...)...)
1920

2021
function broadcasted_index(oldsize, newindex)
2122
index = ones(Int, length(oldsize))

0 commit comments

Comments
 (0)