init for redfun and & or

#24
mcabbott · Aug 17, 2020 · 396b3b6 · 396b3b6 · mcabbott · Aug 17, 2020
1 parent 9d54a24
commit 396b3b6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/macro.jl b/src/macro.jl
@@ -695,6 +695,8 @@ function action_functions(store)
     init = store.redfun == :* ? :(one($TYP)) :
         store.redfun == :max ? :(typemin($TYP)) :
         store.redfun == :min ? :(typemax($TYP)) :
+        store.redfun == :& ? :(true) :
+        store.redfun == :| ? :(false) :
         :(zero($TYP))
 
     # Right now this would allow *= only with reduction * too. Could separate them:

diff --git a/test/parsing.jl b/test/parsing.jl
@@ -383,6 +383,10 @@ end
     @test maximum(A) == @tullio (max) m := float(A[i])
     @test minimum(A) == @tullio (min) m := float(A[i]) # fails with @avx
 
+    @test true == @tullio (&) p := A[i] > 0
+    @test_broken true === @tullio (&) p := A[i] > 0 # sum([true]) isa Int
+    @test true == @tullio (|) q := A[i] > 50
+
     # in-place
     C = copy(A)
     @test cumprod(A) == @tullio (*) C[k] = ifelse(i<=k, A[i], 1)