archive/ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2025-08-25 22:14:37 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Implement initial side exit

Browse source
This commit is contained in:
Takashi Kokubun 2022-12-23 10:43:18 -08:00
parent e750e1e3ee
commit b99d62bf92
3 changed files with 101 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

78
lib/ruby_vm/mjit/compiler.rb
View file

@ -4,9 +4,13 @@ require 'mjit/x86_assembler'
module RubyVM::MJIT
# Compilation status
KeepCompiling = :keep_compiling
CantCompile = :cant_compile
EndBlock = :end_block
KeepCompiling = :KeepCompiling
CantCompile = :CantCompile
EndBlock = :EndBlock
# Ruby constants
Qnil = Fiddle::Qnil
Qundef = Fiddle::Qundef
class Compiler
attr_accessor :write_pos
@ -36,32 +40,6 @@ module RubyVM::MJIT
private
# ec: rdi
# cfp: rsi
def compile_prologue(asm)
asm.mov(:rbx, [:rsi, C.rb_control_frame_t.offsetof(:sp)]) # rbx = cfp->sp
end
def compile_block(asm, iseq)
index = 0
while index < iseq.body.iseq_size
insn = decode_insn(iseq.body.iseq_encoded[index])
status = compile_insn(asm, insn)
if status == EndBlock
break
end
index += insn.len
end
end
def compile_insn(asm, insn)
case insn.name
when :putnil then @insn_compiler.putnil(asm)
when :leave then @insn_compiler.leave(asm)
else raise NotImplementedError, "insn '#{insn.name}' is not supported yet"
end
end
def compile(asm)
start_addr = write_addr
@ -76,6 +54,48 @@ module RubyVM::MJIT
start_addr
end
# ec: rdi
# cfp: rsi
def compile_prologue(asm)
asm.mov(:rbx, [:rsi, C.rb_control_frame_t.offsetof(:sp)]) # rbx = cfp->sp
end
def compile_block(asm, iseq)
index = 0
while index < iseq.body.iseq_size
insn = decode_insn(iseq.body.iseq_encoded[index])
case compile_insn(asm, insn)
when EndBlock
break
when CantCompile
compile_exit(asm, (iseq.body.iseq_encoded + index).to_i)
break
end
index += insn.len
end
end
def compile_insn(asm, insn)
case insn.name
when :putnil then @insn_compiler.putnil(asm)
#when :leave then @insn_compiler.leave(asm)
else CantCompile
end
end
def compile_exit(asm, exit_pc)
# update pc
asm.mov(:rax, exit_pc) # rax = exit_pc
asm.mov([:rsi, C.rb_control_frame_t.offsetof(:pc)], :rax) # cfp->pc = rax
# update sp (TODO: consider JIT state)
asm.add(:rbx, C.VALUE.size) # rbx += 1
asm.mov([:rsi, C.rb_control_frame_t.offsetof(:sp)], :rbx) # cfp->sp = rbx
asm.mov(:rax, Qundef)
asm.ret
end
def decode_insn(encoded)
INSNS.fetch(C.rb_vm_insn_decode(encoded))
end