perf: pre-compute EQ, NEQ, UNM, and BNOT with literals

heir-lang · Jan 21, 2025 · 263bf44 · 263bf44
1 parent a5953a5
commit 263bf44
Showing 1 changed file with 42 additions and 0 deletions.
diff --git a/Heir/CodeGeneration/BytecodeOptimizer.cs b/Heir/CodeGeneration/BytecodeOptimizer.cs
@@ -36,6 +36,25 @@ public List<Instruction> Optimize()
                 // pre-compute the result of operations using only literal values
                 // FUCK THIS CODE
 
+                // unary
+                {
+                    if (instruction.Operand is long or ulong or int or uint or short or ushort or byte or sbyte or double or float or decimal &&
+                        PeekBytecode(1) is
+                        {
+                            OpCode: OpCode.UNM or OpCode.BNOT
+                        } operation)
+                    {
+                        var result = operation.OpCode == OpCode.UNM
+                            ? -Convert.ToDouble(instruction.Operand)
+                            : ~Convert.ToInt64(instruction.Operand);
+
+                        Advance();
+                        Advance();
+                        return instruction.WithOperand(result);
+                    }
+                }
+
+                // binary
                 {
                     if (instruction.Operand is long or ulong or int or uint or short or ushort or byte or sbyte or double or float or decimal &&
                         PeekBytecode(1) is
@@ -116,6 +135,29 @@ public List<Instruction> Optimize()
                         return instruction.WithOperand(result);
                     }
                 }
+                {
+                    if (instruction.Operand is var left &&
+                        PeekBytecode(1) is
+                        {
+                            OpCode: OpCode.PUSH,
+                            Operand: var right
+                        } &&
+                        PeekBytecode(2) is
+                        {
+                            OpCode: OpCode.EQ or OpCode.NEQ
+                        } operation)
+                    {
+                        var equalityComparer = EqualityComparer<object>.Default;
+                        var result = equalityComparer.Equals(left, right);
+                        if (operation.OpCode == OpCode.NEQ)
+                            result = !result;
+
+                        Advance();
+                        Advance();
+                        Advance();
+                        return instruction.WithOperand(result);
+                    }
+                }
 
                 break;
             }