Skip to content

Commit 3120724

Browse files
committed
Fixup NonExtruded indexing
1 parent c54b40b commit 3120724

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

base/broadcast.jl

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -918,7 +918,10 @@ struct _NonExtruded28126{T}
918918
x::T
919919
end
920920
@inline axes(b::_NonExtruded28126) = axes(b.x)
921-
Base.@propagate_inbounds _broadcast_getindex(b::_NonExtruded28126, i) = b.x[i]
921+
Base.@propagate_inbounds _broadcast_getindex(b::_NonExtruded28126, i) = _broadcast_getindex(b, i)
922+
Base.@propagate_inbounds _broadcast_getindex(b::_NonExtruded28126{<:AbstractArray{<:Any,0}}, i) = b.x[]
923+
Base.@propagate_inbounds _broadcast_getindex(b::_NonExtruded28126{<:AbstractVector}, i) = b.x[i[1]]
924+
Base.@propagate_inbounds _broadcast_getindex(b::_NonExtruded28126{<:AbstractArray}, i) = b.x[i]
922925
_nonextrude_28126(x::AbstractArray) = _NonExtruded28126(x)
923926
_nonextrude_28126(x) = x
924927

0 commit comments

Comments
 (0)