optimize /×⧈≍ as all same

uiua-lang · Dec 28, 2024 · 07cbce7 · 07cbce7
1 parent d2464e0
commit 07cbce7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/compile/optimize.rs b/src/compile/optimize.rs
@@ -368,6 +368,16 @@ impl Optimization for AllSameOpt {
             [ImplPrim(CountUnique, span), Push(val), Prim(Le, _), ..] if *val == 1 => {
                 Some((3, ImplPrim(AllSame, *span)))
             }
+            [Mod(Stencil, stencil_args, span), Mod(Reduce, reduce_args, _), ..] => {
+                let ([stencil_f], [reduce_f]) = (stencil_args.as_slice(), reduce_args.as_slice())
+                else {
+                    return None;
+                };
+                match (&stencil_f.node, &reduce_f.node) {
+                    (Prim(Match, _), Prim(Mul, _)) => Some((2, ImplPrim(AllSame, *span))),
+                    _ => None,
+                }
+            }
             _ => None,
         })
     }

diff --git a/tests/optimized.ua b/tests/optimized.ua
@@ -214,6 +214,12 @@ F ← ⬚10(/+◌1⊞(×¯))
 ⍤⤙≍ [≍∘⊸↻1] ⟜(◴[⊃(≤1⧻◴|≍⊸↻1|≍⊸(↻1))]) [1_2 1_2]
 ⍤⤙≍ [≍∘⊸↻1] ⟜(◴[⊃(≤1⧻◴|≍⊸↻1|≍⊸(↻1))]) [1_2 3_4]
 ⍤⤙≍ [≍∘⊸↻1] ⟜(◴[⊃(≤1⧻◴|≍⊸↻1|≍⊸(↻1))]) []
+⍤⤙≍ ⊃(/×∘⧈≍|/×⧈≍) [1 2 3]
+⍤⤙≍ ⊃(/×∘⧈≍|/×⧈≍) [1_2 3_4 5_6]
+⍤⤙≍ ⊃(/×∘⧈≍|/×⧈≍) [1 1]
+⍤⤙≍ ⊃(/×∘⧈≍|/×⧈≍) [1 0]
+⍤⤙≍ ⊃(/×∘⧈≍|/×⧈≍) [1]
+⍤⤙≍ ⊃(/×∘⧈≍|/×⧈≍) []
 
 # Tuples
 ⍤⤙≍ ⊃⧅≠⧅(∘≠) 3 ⇡4