From 6d59cac4ad8944a725762165d470953dd534528b Mon Sep 17 00:00:00 2001
From: Takashi Kokubun <takashikkbn@gmail.com>
Date: Thu, 2 Mar 2023 23:28:24 -0800
Subject: Optimize Integer#[]

---
 lib/ruby_vm/mjit/insn_compiler.rb | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'lib/ruby_vm')

diff --git a/lib/ruby_vm/mjit/insn_compiler.rb b/lib/ruby_vm/mjit/insn_compiler.rb
index e68c68e3a9..38ba71a964 100644
--- a/lib/ruby_vm/mjit/insn_compiler.rb
+++ b/lib/ruby_vm/mjit/insn_compiler.rb
@@ -1778,6 +1778,29 @@ module RubyVM::MJIT
       true
     end
 
+    # @param jit [RubyVM::MJIT::JITState]
+    # @param ctx [RubyVM::MJIT::Context]
+    # @param asm [RubyVM::MJIT::Assembler]
+    def jit_rb_int_aref(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)
+
+      asm.comment('rb_int_aref')
+      y_opnd = ctx.stack_pop
+      x_opnd = ctx.stack_pop
+
+      asm.mov(C_ARGS[0], x_opnd)
+      asm.mov(C_ARGS[1], y_opnd)
+      asm.call(C.rb_fix_aref)
+
+      ret_opnd = ctx.stack_push
+      asm.mov(ret_opnd, C_RET)
+      true
+    end
+
     # @param jit [RubyVM::MJIT::JITState]
     # @param ctx [RubyVM::MJIT::Context]
     # @param asm [RubyVM::MJIT::Assembler]
@@ -1810,6 +1833,7 @@ module RubyVM::MJIT
       register_cfunc_method(Integer, :===, :jit_rb_int_equal)
       register_cfunc_method(Integer, :*, :jit_rb_int_mul)
       register_cfunc_method(Integer, :/, :jit_rb_int_div)
+      register_cfunc_method(Integer, :[], :jit_rb_int_aref)
 
       register_cfunc_method(Array, :<<, :jit_rb_ary_push)
     end
-- 
cgit v1.2.3