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

YJIT: Add codegen for Array#<< (#7645)

Browse Source
This commit is contained in:
Takashi Kokubun 2023-04-03 14:10:35 -07:00 committed by GitHub
parent ba4ff2552e
commit 38209ffdca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
Notes: git 2023-04-03 21:10:58 +00:00 
Merged-By: k0kubun <takashikkbn@gmail.com>
3 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
yjit/bindgen/src/main.rs
View File

@ -136,6 +136,7 @@ fn main() {
.allowlist_function("rb_ary_resurrect") .allowlist_function("rb_ary_resurrect")
.allowlist_function("rb_ary_clear") .allowlist_function("rb_ary_clear")
.allowlist_function("rb_ary_dup") .allowlist_function("rb_ary_dup")
.allowlist_function("rb_ary_push")
.allowlist_function("rb_ary_unshift_m") .allowlist_function("rb_ary_unshift_m")
.allowlist_function("rb_yjit_rb_ary_subseq_length") .allowlist_function("rb_yjit_rb_ary_subseq_length")

26
yjit/src/codegen.rs
View File

@ -4542,6 +4542,31 @@ fn jit_rb_ary_empty_p(
return true; return true;
} }
fn jit_rb_ary_push(
jit: &mut JITState,
ctx: &mut Context,
asm: &mut Assembler,
_ocb: &mut OutlinedCb,
_ci: *const rb_callinfo,
_cme: *const rb_callable_method_entry_t,
_block: Option<IseqPtr>,
_argc: i32,
_known_recv_class: *const VALUE,
) -> bool {
asm.comment("Array#<<");
// rb_ary_push allocates memory for buffer extension
jit_prepare_routine_call(jit, ctx, asm);
let item_opnd = ctx.stack_pop(1);
let ary_opnd = ctx.stack_pop(1);
let ret = asm.ccall(rb_ary_push as *const u8, vec![ary_opnd, item_opnd]);
let ret_opnd = ctx.stack_push(Type::TArray);
asm.mov(ret_opnd, ret);
true
}
fn jit_obj_respond_to( fn jit_obj_respond_to(
jit: &mut JITState, jit: &mut JITState,
ctx: &mut Context, ctx: &mut Context,
@ -8071,6 +8096,7 @@ impl CodegenGlobals {
// rb_ary_empty_p() method in array.c // rb_ary_empty_p() method in array.c
self.yjit_reg_method(rb_cArray, "empty?", jit_rb_ary_empty_p); self.yjit_reg_method(rb_cArray, "empty?", jit_rb_ary_empty_p);
self.yjit_reg_method(rb_cArray, "<<", jit_rb_ary_push);
self.yjit_reg_method(rb_mKernel, "respond_to?", jit_obj_respond_to); self.yjit_reg_method(rb_mKernel, "respond_to?", jit_obj_respond_to);
self.yjit_reg_method(rb_mKernel, "block_given?", jit_rb_f_block_given_p); self.yjit_reg_method(rb_mKernel, "block_given?", jit_rb_f_block_given_p);

1
yjit/src/cruby_bindings.inc.rs
View File

@ -1094,6 +1094,7 @@ extern "C" {
pub fn rb_ary_store(ary: VALUE, key: ::std::os::raw::c_long, val: VALUE); pub fn rb_ary_store(ary: VALUE, key: ::std::os::raw::c_long, val: VALUE);
pub fn rb_ary_dup(ary: VALUE) -> VALUE; pub fn rb_ary_dup(ary: VALUE) -> VALUE;
pub fn rb_ary_resurrect(ary: VALUE) -> VALUE; pub fn rb_ary_resurrect(ary: VALUE) -> VALUE;
pub fn rb_ary_push(ary: VALUE, elem: VALUE) -> VALUE;
pub fn rb_ary_clear(ary: VALUE) -> VALUE; pub fn rb_ary_clear(ary: VALUE) -> VALUE;
pub fn rb_hash_new() -> VALUE; pub fn rb_hash_new() -> VALUE;
pub fn rb_hash_aref(hash: VALUE, key: VALUE) -> VALUE; pub fn rb_hash_aref(hash: VALUE, key: VALUE) -> VALUE;