help type inference for logical indexing (#29633)

(cherry picked from commit aa72f72)
JuliaLang · Feb 20, 2020 · 430df6e · 430df6e
1 parent 6d3120c
commit 430df6e
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)