archive/ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2025-08-27 15:06:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Optimize Integer#===

Browse source
This commit is contained in:
Takashi Kokubun 2023-03-02 22:52:32 -08:00
parent 92efd0569a
commit 2ecf77ce20
Notes: git 2023-03-06 07:29:45 +00:00
1 changed files with 30 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

32
lib/ruby_vm/mjit/insn_compiler.rb
View file

@ -1686,7 +1686,7 @@ module RubyVM::MJIT
# @param asm [RubyVM::MJIT::Assembler] # @param asm [RubyVM::MJIT::Assembler]
def jit_rb_obj_not(jit, ctx, asm, argc) def jit_rb_obj_not(jit, ctx, asm, argc)
return false if argc != 0 return false if argc != 0
asm.comment('jit_rb_obj_not') asm.comment('rb_obj_not')
recv = ctx.stack_pop recv = ctx.stack_pop
# This `test` sets ZF only for Qnil and Qfalse, which let cmovz set. # This `test` sets ZF only for Qnil and Qfalse, which let cmovz set.
@ -1699,13 +1699,38 @@ module RubyVM::MJIT
true true
end 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 jit [RubyVM::MJIT::JITState]
# @param ctx [RubyVM::MJIT::Context] # @param ctx [RubyVM::MJIT::Context]
# @param asm [RubyVM::MJIT::Assembler] # @param asm [RubyVM::MJIT::Assembler]
def jit_rb_int_mul(jit, ctx, asm, argc) def jit_rb_int_mul(jit, ctx, asm, argc)
return false if argc != 1 return false if argc != 1
return false unless two_fixnums_on_stack?(jit) 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) side_exit = side_exit(jit, ctx)
guard_two_fixnums(jit, ctx, asm, side_exit) guard_two_fixnums(jit, ctx, asm, side_exit)
@ -1730,6 +1755,9 @@ module RubyVM::MJIT
def register_cfunc_codegen_funcs def register_cfunc_codegen_funcs
register_cfunc_method(BasicObject, :!, :jit_rb_obj_not) 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) register_cfunc_method(Integer, :*, :jit_rb_int_mul)
end end