From 2ecf77ce205d9732c8c1fcfbf031835c4d53ebbd Mon Sep 17 00:00:00 2001
From: Takashi Kokubun <takashikkbn@gmail.com>
Date: Thu, 2 Mar 2023 22:52:32 -0800
Subject: [PATCH] Optimize Integer#===

---
 lib/ruby_vm/mjit/insn_compiler.rb | 32 +++++++++++++++++++++++++++++--
 1 file changed, 30 insertions(+), 2 deletions(-)

diff --git a/lib/ruby_vm/mjit/insn_compiler.rb b/lib/ruby_vm/mjit/insn_compiler.rb
index 42f95cee1e..6141b6de8d 100644
--- a/lib/ruby_vm/mjit/insn_compiler.rb
+++ b/lib/ruby_vm/mjit/insn_compiler.rb
@@ -1686,7 +1686,7 @@ module RubyVM::MJIT
     # @param asm [RubyVM::MJIT::Assembler]
     def jit_rb_obj_not(jit, ctx, asm, argc)
       return false if argc != 0
-      asm.comment('jit_rb_obj_not')
+      asm.comment('rb_obj_not')
 
       recv = ctx.stack_pop
       # This `test` sets ZF only for Qnil and Qfalse, which let cmovz set.
@@ -1699,13 +1699,38 @@ module RubyVM::MJIT
       true
     end
 
+    # @param jit [RubyVM::MJIT::JITState]
+    # @param ctx [RubyVM::MJIT::Context]
+    # @param asm [RubyVM::MJIT::Assembler]
+    def jit_rb_int_equal(jit, ctx, asm, argc)
+      return false if argc != 1
+      return false unless two_fixnums_on_stack?(jit)
+
+      side_exit = side_exit(jit, ctx)
+      guard_two_fixnums(jit, ctx, asm, side_exit)
+
+      # Compare the arguments
+      asm.comment('rb_int_equal')
+      arg1 = ctx.stack_pop(1)
+      arg0 = ctx.stack_pop(1)
+      asm.mov(:rax, arg1)
+      asm.cmp(arg0, :rax)
+      asm.mov(:rax, Qfalse)
+      asm.mov(:rcx, Qtrue)
+      asm.cmove(:rax, :rcx)
+
+      stack_ret = ctx.stack_push
+      asm.mov(stack_ret, :rax)
+      true
+    end
+
     # @param jit [RubyVM::MJIT::JITState]
     # @param ctx [RubyVM::MJIT::Context]
     # @param asm [RubyVM::MJIT::Assembler]
     def jit_rb_int_mul(jit, ctx, asm, argc)
       return false if argc != 1
       return false unless two_fixnums_on_stack?(jit)
-      asm.comment('jit_rb_int_mul')
+      asm.comment('rb_int_mul')
 
       side_exit = side_exit(jit, ctx)
       guard_two_fixnums(jit, ctx, asm, side_exit)
@@ -1730,6 +1755,9 @@ module RubyVM::MJIT
 
     def register_cfunc_codegen_funcs
       register_cfunc_method(BasicObject, :!, :jit_rb_obj_not)
+
+      register_cfunc_method(Integer, :==, :jit_rb_int_equal)
+      register_cfunc_method(Integer, :===, :jit_rb_int_equal)
       register_cfunc_method(Integer, :*, :jit_rb_int_mul)
     end