help type inference for logical indexing (#29633)

JuliaLang · Oct 29, 2018 · aa72f72 · aa72f72 · nanosoldier · Oct 30, 2018
1 parent cd0bd33
commit aa72f72
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/base/multidimensional.jl b/base/multidimensional.jl
@@ -517,6 +517,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)