1berry/ruby
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Traverse bytecode using block queue

Browse Source
This commit is contained in:
Max Bernstein 2025-02-10 11:10:21 -05:00 committed by Takashi Kokubun
parent 3b97bcea13
commit 948560ac5a
Notes: git 2025-04-18 13:49:14 +00:00 
Merged: https://github.com/ruby/ruby/pull/13131
1 changed files with 154 additions and 156 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
zjit/src/ir.rs
View File

@ -313,14 +313,6 @@ pub enum ParseError {
pub fn iseq_to_ssa(iseq: *const rb_iseq_t) -> Result<Function, ParseError> {
let mut fun = Function::new(iseq);
// TODO(max): Queue TranslationContexts so that we can assign states to basic blocks instead of
// flowing one state through the linear iseq
let mut state = FrameState::new();
let mut block = fun.entry_block;
let iseq_size = unsafe { get_iseq_encoded_size(iseq) };
let mut insn_idx = 0;
// Compute a map of PC->Block by finding jump targets
let jump_targets = compute_jump_targets(iseq);
let mut insn_idx_to_block = HashMap::new();
@ -331,23 +323,23 @@ pub fn iseq_to_ssa(iseq: *const rb_iseq_t) -> Result<Function, ParseError> {
insn_idx_to_block.insert(insn_idx, fun.new_block());
}
while insn_idx < iseq_size {
// Get the block id for this instruction
if let Some(block_id) = insn_idx_to_block.get(&insn_idx) {
block = *block_id;
}
// Iteratively fill out basic blocks using a queue
// TODO(max): Basic block arguments at edges
let mut queue = std::collections::VecDeque::new();
queue.push_back((FrameState::new(), fun.entry_block, /*insn_idx=*/0 as u32));
let iseq_size = unsafe { get_iseq_encoded_size(iseq) };
while let Some((mut state, block, mut insn_idx)) = queue.pop_front() {
while insn_idx < iseq_size {
// Get the current pc and opcode
let pc = unsafe { rb_iseq_pc_at_idx(iseq, insn_idx.into()) };
state.pc = unsafe { *pc };
fun.push_insn(block, Insn::Snapshot { state: state.clone() });
// try_into() call below is unfortunate. Maybe pick i32 instead of usize for opcodes.
let opcode: u32 = unsafe { rb_iseq_opcode_at_pc(iseq, pc) }
.try_into()
.unwrap();
// Move to the next instruction to compile
insn_idx += insn_len(opcode as usize);
@ -488,6 +480,12 @@ pub fn iseq_to_ssa(iseq: *const rb_iseq_t) -> Result<Function, ParseError> {
}
_ => eprintln!("zjit: to_ssa: unknown opcode `{}'", insn_name(opcode as usize)),
}
if insn_idx_to_block.contains_key(&insn_idx) {
queue.push_back((state, insn_idx_to_block[&insn_idx], insn_idx));
break;
}
}
}
let formatter = FunctionPrinter::from(&fun);