fix a bug in a fork optimization

uiua-lang · Nov 6, 2024 · 540e419 · 540e419
1 parent 43e8100
commit 540e419
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/compile/modifier.rs b/src/compile/modifier.rs
@@ -453,9 +453,10 @@ impl Compiler {
                     );
                 }
                 let span = self.add_span(modified.modifier.span.clone());
-                match f.sig.args {
-                    0 => Node::from_iter([g.node, f.node]),
-                    1 => Node::from_iter([Node::Mod(On, eco_vec![g], span), f.node]),
+                match (f.sig.args, g.sig.args) {
+                    (0, _) => Node::from_iter([g.node, f.node]),
+                    (1, 0) => Node::from_iter([Node::Mod(Dip, eco_vec![g], span), f.node]),
+                    (1, _) => Node::from_iter([Node::Mod(On, eco_vec![g], span), f.node]),
                     _ => Node::Mod(Fork, eco_vec![f, g], span),
                 }
             }

diff --git a/tests/optimized.ua b/tests/optimized.ua
@@ -81,6 +81,10 @@
 ⍤⤙≍ [⊞⊃+- .⇡5] [⊞⊃(+|-) .⇡5]
 ⍤⤙≍ {⊞(∘⊃(+|++))..⇡2} {⊞⊃(+|++)..⇡2}
 
+# Fork reductions
+⍤⤙≍ [5 ¯3] [⊃5¯] 3
+⍤⤙≍ [¯3 5] [⊃¯ 5] 3
+
 ⍤⤙≍ [0 0 0] ≡⋅0 ⇡3
 ⍤⤙≍ [0 0 0] ∵⋅0 ⇡3
 ⍤⤙≍ [0 0 0] ≡⋅0 ↯3_3 0