зеркало из https://github.com/github/ruby.git
Implement opt_aref for Hash
This commit is contained in:
Родитель
a458923fe5
Коммит
8d099ff699
|
@ -18,5 +18,13 @@ module RubyVM::MJIT
|
|||
self.sp_offset -= size
|
||||
[SP, C.VALUE.size * self.sp_offset]
|
||||
end
|
||||
|
||||
def stack_opnd(depth_from_top)
|
||||
[SP, C.VALUE.size * (self.sp_offset - 1 - depth_from_top)]
|
||||
end
|
||||
|
||||
def sp_opnd(offset_bytes)
|
||||
[SP, (C.VALUE.size * self.sp_offset) + offset_bytes]
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -18,7 +18,7 @@ module RubyVM::MJIT
|
|||
asm.incr_counter(:mjit_insns_count)
|
||||
asm.comment("Insn: #{insn.name}")
|
||||
|
||||
# 15/101
|
||||
# 16/101
|
||||
case insn.name
|
||||
when :nop then nop(jit, ctx, asm)
|
||||
# getlocal
|
||||
|
@ -102,7 +102,7 @@ module RubyVM::MJIT
|
|||
# opt_ltlt
|
||||
# opt_and
|
||||
# opt_or
|
||||
# opt_aref
|
||||
when :opt_aref then opt_aref(jit, ctx, asm)
|
||||
# opt_aset
|
||||
# opt_aset_with
|
||||
# opt_aref_with
|
||||
|
@ -488,7 +488,70 @@ module RubyVM::MJIT
|
|||
# opt_ltlt
|
||||
# opt_and
|
||||
# opt_or
|
||||
# opt_aref
|
||||
|
||||
# @param jit [RubyVM::MJIT::JITState]
|
||||
# @param ctx [RubyVM::MJIT::Context]
|
||||
# @param asm [RubyVM::MJIT::Assembler]
|
||||
def opt_aref(jit, ctx, asm)
|
||||
cd = C.rb_call_data.new(jit.operand(0))
|
||||
argc = C.vm_ci_argc(cd.ci)
|
||||
|
||||
if argc != 1
|
||||
asm.incr_counter(:optaref_argc_not_one)
|
||||
return CantCompile
|
||||
end
|
||||
|
||||
unless jit.at_current_insn?
|
||||
defer_compilation(jit, ctx, asm)
|
||||
return EndBlock
|
||||
end
|
||||
|
||||
comptime_recv = jit.peek_at_stack(1)
|
||||
comptime_obj = jit.peek_at_stack(0)
|
||||
|
||||
side_exit = side_exit(jit, ctx)
|
||||
|
||||
if comptime_recv.class == Array && fixnum?(comptime_obj)
|
||||
asm.incr_counter(:optaref_array)
|
||||
CantCompile
|
||||
elsif comptime_recv.class == Hash
|
||||
unless @invariants.assume_bop_not_redefined(jit, C.HASH_REDEFINED_OP_FLAG, C.BOP_AREF)
|
||||
return CantCompile
|
||||
end
|
||||
|
||||
recv_opnd = ctx.stack_opnd(1)
|
||||
|
||||
# Guard that the receiver is a Hash
|
||||
not_hash_exit = counted_exit(side_exit, :optaref_not_hash)
|
||||
if jit_guard_known_class(jit, ctx, asm, comptime_recv.class, recv_opnd, comptime_recv, not_hash_exit) == CantCompile
|
||||
return CantCompile
|
||||
end
|
||||
|
||||
# Prepare to call rb_hash_aref(). It might call #hash on the key.
|
||||
jit_prepare_routine_call(jit, ctx, asm)
|
||||
|
||||
asm.comment('call rb_hash_aref')
|
||||
key_opnd = ctx.stack_opnd(0)
|
||||
recv_opnd = ctx.stack_opnd(1)
|
||||
asm.mov(:rdi, recv_opnd)
|
||||
asm.mov(:rsi, key_opnd)
|
||||
asm.call(C.rb_hash_aref)
|
||||
|
||||
# Pop the key and the receiver
|
||||
ctx.stack_pop(2)
|
||||
|
||||
stack_ret = ctx.stack_push
|
||||
asm.mov(stack_ret, :rax)
|
||||
|
||||
# Let guard chains share the same successor
|
||||
jump_to_next_insn(jit, ctx, asm)
|
||||
EndBlock
|
||||
else
|
||||
asm.incr_counter(:optaref_send)
|
||||
CantCompile
|
||||
end
|
||||
end
|
||||
|
||||
# opt_aset
|
||||
# opt_aset_with
|
||||
# opt_aref_with
|
||||
|
@ -665,6 +728,35 @@ module RubyVM::MJIT
|
|||
end
|
||||
end
|
||||
|
||||
# @param jit [RubyVM::MJIT::JITState]
|
||||
# @param ctx [RubyVM::MJIT::Context]
|
||||
# @param asm [RubyVM::MJIT::Assembler]
|
||||
def jit_prepare_routine_call(jit, ctx, asm)
|
||||
jit_save_pc(jit, asm)
|
||||
jit_save_sp(jit, ctx, asm)
|
||||
end
|
||||
|
||||
# @param jit [RubyVM::MJIT::JITState]
|
||||
# @param asm [RubyVM::MJIT::Assembler]
|
||||
def jit_save_pc(jit, asm)
|
||||
next_pc = jit.pc + jit.insn.len * C.VALUE.size # Use the next one for backtrace and side exits
|
||||
asm.comment('save PC to CFP')
|
||||
asm.mov(:rax, next_pc)
|
||||
asm.mov([CFP, C.rb_control_frame_t.offsetof(:pc)], :rax)
|
||||
end
|
||||
|
||||
# @param jit [RubyVM::MJIT::JITState]
|
||||
# @param ctx [RubyVM::MJIT::Context]
|
||||
# @param asm [RubyVM::MJIT::Assembler]
|
||||
def jit_save_sp(jit, ctx, asm)
|
||||
if ctx.sp_offset != 0
|
||||
asm.comment('save SP to CFP')
|
||||
asm.lea(SP, ctx.sp_opnd(0))
|
||||
asm.mov([CFP, C.rb_control_frame_t.offsetof(:sp)], SP)
|
||||
ctx.sp_offset = 0
|
||||
end
|
||||
end
|
||||
|
||||
# @param jit [RubyVM::MJIT::JITState]
|
||||
# @param ctx [RubyVM::MJIT::Context]
|
||||
# @param asm [RubyVM::MJIT::Assembler]
|
||||
|
|
|
@ -36,6 +36,7 @@ module RubyVM::MJIT
|
|||
|
||||
print_counters(stats, prefix: 'send_', prompt: 'method call exit reasons')
|
||||
print_counters(stats, prefix: 'getivar_', prompt: 'getinstancevariable exit reasons')
|
||||
print_counters(stats, prefix: 'optaref_', prompt: 'opt_aref exit reasons')
|
||||
|
||||
$stderr.puts "compiled_block_count: #{format('%10d', stats[:compiled_block_count])}"
|
||||
$stderr.puts "side_exit_count: #{format('%10d', stats[:side_exit_count])}"
|
||||
|
|
5
mjit_c.h
5
mjit_c.h
|
@ -132,6 +132,11 @@ MJIT_RUNTIME_COUNTERS(
|
|||
getivar_special_const,
|
||||
getivar_too_complex,
|
||||
|
||||
optaref_argc_not_one,
|
||||
optaref_array,
|
||||
optaref_not_hash,
|
||||
optaref_send,
|
||||
|
||||
compiled_block_count
|
||||
)
|
||||
#undef MJIT_RUNTIME_COUNTERS
|
||||
|
|
16
mjit_c.rb
16
mjit_c.rb
|
@ -127,6 +127,10 @@ module RubyVM::MJIT # :nodoc: all
|
|||
Primitive.cexpr! 'RBOOL(FL_TEST_RAW((VALUE)NUM2SIZET(_value), (VALUE)NUM2SIZET(flags)))'
|
||||
end
|
||||
|
||||
def rb_hash_aref
|
||||
Primitive.cexpr! 'SIZET2NUM((size_t)rb_hash_aref)'
|
||||
end
|
||||
|
||||
#========================================================================================
|
||||
#
|
||||
# Old stuff
|
||||
|
@ -287,6 +291,10 @@ module RubyVM::MJIT # :nodoc: all
|
|||
Primitive.cexpr! %q{ INT2NUM(VM_ENV_DATA_INDEX_SPECVAL) }
|
||||
end
|
||||
|
||||
def C.BOP_AREF
|
||||
Primitive.cexpr! %q{ UINT2NUM(BOP_AREF) }
|
||||
end
|
||||
|
||||
def C.BOP_LT
|
||||
Primitive.cexpr! %q{ UINT2NUM(BOP_LT) }
|
||||
end
|
||||
|
@ -299,6 +307,10 @@ module RubyVM::MJIT # :nodoc: all
|
|||
Primitive.cexpr! %q{ UINT2NUM(BOP_PLUS) }
|
||||
end
|
||||
|
||||
def C.HASH_REDEFINED_OP_FLAG
|
||||
Primitive.cexpr! %q{ UINT2NUM(HASH_REDEFINED_OP_FLAG) }
|
||||
end
|
||||
|
||||
def C.INTEGER_REDEFINED_OP_FLAG
|
||||
Primitive.cexpr! %q{ UINT2NUM(INTEGER_REDEFINED_OP_FLAG) }
|
||||
end
|
||||
|
@ -887,6 +899,10 @@ module RubyVM::MJIT # :nodoc: all
|
|||
getivar_not_t_object: [CType::Immediate.parse("size_t"), Primitive.cexpr!("OFFSETOF((*((struct rb_mjit_runtime_counters *)NULL)), getivar_not_t_object)")],
|
||||
getivar_special_const: [CType::Immediate.parse("size_t"), Primitive.cexpr!("OFFSETOF((*((struct rb_mjit_runtime_counters *)NULL)), getivar_special_const)")],
|
||||
getivar_too_complex: [CType::Immediate.parse("size_t"), Primitive.cexpr!("OFFSETOF((*((struct rb_mjit_runtime_counters *)NULL)), getivar_too_complex)")],
|
||||
optaref_argc_not_one: [CType::Immediate.parse("size_t"), Primitive.cexpr!("OFFSETOF((*((struct rb_mjit_runtime_counters *)NULL)), optaref_argc_not_one)")],
|
||||
optaref_array: [CType::Immediate.parse("size_t"), Primitive.cexpr!("OFFSETOF((*((struct rb_mjit_runtime_counters *)NULL)), optaref_array)")],
|
||||
optaref_not_hash: [CType::Immediate.parse("size_t"), Primitive.cexpr!("OFFSETOF((*((struct rb_mjit_runtime_counters *)NULL)), optaref_not_hash)")],
|
||||
optaref_send: [CType::Immediate.parse("size_t"), Primitive.cexpr!("OFFSETOF((*((struct rb_mjit_runtime_counters *)NULL)), optaref_send)")],
|
||||
compiled_block_count: [CType::Immediate.parse("size_t"), Primitive.cexpr!("OFFSETOF((*((struct rb_mjit_runtime_counters *)NULL)), compiled_block_count)")],
|
||||
)
|
||||
end
|
||||
|
|
|
@ -348,9 +348,11 @@ generator = BindingGenerator.new(
|
|||
VM_ENV_DATA_INDEX_SPECVAL
|
||||
],
|
||||
UINT: %w[
|
||||
BOP_AREF
|
||||
BOP_LT
|
||||
BOP_MINUS
|
||||
BOP_PLUS
|
||||
HASH_REDEFINED_OP_FLAG
|
||||
INTEGER_REDEFINED_OP_FLAG
|
||||
METHOD_VISI_PRIVATE
|
||||
METHOD_VISI_PROTECTED
|
||||
|
|
Загрузка…
Ссылка в новой задаче