archive/ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2025-08-15 13:39:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

YJIT: Filter & calls from specialized C method codegen

Browse source 
Evident with the crash reported in [Bug #20997], the C replacement
codegen functions aren't authored to handle block arguments (nor
should they because the extra code from the complexity defeats
optimization). Filter sites with VM_CALL_ARGS_BLOCKARG.
This commit is contained in:
Alan Wu 2025-01-08 19:07:07 -05:00 • committed by Takashi Kokubun
parent 706f1d0573
commit 294aef53b8
2 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
bootstraptest/test_yjit.rb
View file

@ -5379,3 +5379,11 @@ assert_equal '["x", "Y", "c", "A", "t", "A", "b", "C", "d"]', <<~'RUBY'
Swap.new("xy").swap + Swap.new("cat").reverse_odd + Swap.new("abcd").reverse_even
RUBY
assert_normal_exit %{
class Bug20997
def foo(&) = self.class.name(&)
new.foo
end
}

3
yjit/src/codegen.rs
View file

@ -6894,11 +6894,12 @@ fn gen_send_cfunc(
// Increment total cfunc send count
gen_counter_incr(jit, asm, Counter::num_send_cfunc);
// Delegate to codegen for C methods if we have it.
// Delegate to codegen for C methods if we have it and the callsite is simple enough.
if kw_arg.is_null() &&
!kw_splat &&
flags & VM_CALL_OPT_SEND == 0 &&
flags & VM_CALL_ARGS_SPLAT == 0 &&
flags & VM_CALL_ARGS_BLOCKARG == 0 &&
(cfunc_argc == -1 || argc == cfunc_argc) {
let expected_stack_after = asm.ctx.get_stack_size() as i32 - argc;
if let Some(known_cfunc_codegen) = lookup_cfunc_codegen(unsafe { (*cme).def }) {