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

YJIT: implement opt_newarray_min YARV instruction (#6888)

Browse Source
This commit is contained in:
Maxime Chevalier-Boisvert 2022-12-08 17:31:33 -05:00 committed by GitHub
parent c9076d546a
commit b26c9ce5e9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
Notes: git 2022-12-08 22:31:52 +00:00 
Merged-By: maximecb <maximecb@ruby-lang.org>
2 changed files with 43 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vm_insnhelper.c
View File

@ -5216,6 +5216,12 @@ vm_opt_newarray_min(rb_execution_context_t *ec, rb_num_t num, const VALUE *ptr)
}
}
VALUE
rb_vm_opt_newarray_min(rb_execution_context_t *ec, rb_num_t num, const VALUE *ptr)
{
return vm_opt_newarray_min(ec, num, ptr);
}
#undef id_cmp
#define IMEMO_CONST_CACHE_SHAREABLE IMEMO_FL_USER0

37
yjit/src/codegen.rs
View File

@ -3304,6 +3304,42 @@ fn gen_opt_str_uminus(
KeepCompiling
}
fn gen_opt_newarray_min(
jit: &mut JITState,
ctx: &mut Context,
asm: &mut Assembler,
_ocb: &mut OutlinedCb,
) -> CodegenStatus {
let num = jit_get_arg(jit, 0).as_u32();
// Save the PC and SP because we may allocate
jit_prepare_routine_call(jit, ctx, asm);
extern "C" {
fn rb_vm_opt_newarray_min(ec: EcPtr, num: u32, elts: *const VALUE) -> VALUE;
}
let offset_magnitude = (SIZEOF_VALUE as u32) * num;
let values_opnd = ctx.sp_opnd(-(offset_magnitude as isize));
let values_ptr = asm.lea(values_opnd);
let val_opnd = asm.ccall(
rb_vm_opt_newarray_min as *const u8,
vec![
EC,
num.into(),
values_ptr
],
);
ctx.stack_pop(num.as_usize());
let stack_ret = ctx.stack_push(Type::Unknown);
asm.mov(stack_ret, val_opnd);
KeepCompiling
}
fn gen_opt_not(
jit: &mut JITState,
ctx: &mut Context,
@ -6931,6 +6967,7 @@ fn get_gen_fn(opcode: VALUE) -> Option<InsnGenFn> {
YARVINSN_opt_mod => Some(gen_opt_mod),
YARVINSN_opt_str_freeze => Some(gen_opt_str_freeze),
YARVINSN_opt_str_uminus => Some(gen_opt_str_uminus),
YARVINSN_opt_newarray_min => Some(gen_opt_newarray_min),
YARVINSN_splatarray => Some(gen_splatarray),
YARVINSN_concatarray => Some(gen_concatarray),
YARVINSN_newrange => Some(gen_newrange),