help type inference for logical indexing (#29633)

(cherry picked from commit aa72f72)
JuliaLang · Oct 31, 2018 · c3ed038 · c3ed038
1 parent 7210943
commit c3ed038
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/base/multidimensional.jl b/base/multidimensional.jl
@@ -514,6 +514,10 @@ end
     # We're looking for the n-th true element, using iterator r at state i
     n = s[1]
     n > length(L) && return nothing
+    #unroll once to help inference, cf issue #29418
+    idx, i = iterate(tail(s)...)
+    s = (n+1, s[2], i)
+    L.mask[idx] && return (idx, s)
     while true
         idx, i = iterate(tail(s)...)
         s = (n+1, s[2], i)