ZJIT: Add stat for def_type of send_without_block fallbacks

Author: st0012

zjit.rb

@@ -45,6 +45,7 @@ def stats_string
     print_counters_with_prefix(prefix: 'compile_error_', prompt: 'compile error reasons', buf:, stats:, limit: 20)
     print_counters_with_prefix(prefix: 'exit_', prompt: 'side exit reasons', buf:, stats:, limit: 20)
     print_counters_with_prefix(prefix: 'dynamic_send_type_', prompt: 'dynamic send types', buf:, stats:, limit: 20)
+    print_counters_with_prefix(prefix: 'send_fallback_', prompt: 'send fallback def_types', buf:, stats:, limit: 20)
 
     # Show the most important stats ratio_in_zjit at the end
     print_counters([

zjit/src/codegen.rs

@@ -364,7 +364,10 @@ fn gen_insn(cb: &mut CodeBlock, jit: &mut JITState, asm: &mut Assembler, functio
         Insn::IfTrue { val, target } => no_output!(gen_if_true(jit, asm, opnd!(val), target)),
         Insn::IfFalse { val, target } => no_output!(gen_if_false(jit, asm, opnd!(val), target)),
         &Insn::Send { cd, blockiseq, state, .. } => gen_send(jit, asm, cd, blockiseq, &function.frame_state(state)),
-        Insn::SendWithoutBlock { cd, state, .. } => gen_send_without_block(jit, asm, *cd, &function.frame_state(*state)),
+        Insn::SendWithoutBlock { cd, state, def_type, .. } => {
+            if let Some(def_type) = def_type { update_send_fallback_counter(asm, *def_type); }
+            gen_send_without_block(jit, asm, *cd, &function.frame_state(*state))
+        }
         // Give up SendWithoutBlockDirect for 6+ args since asm.ccall() doesn't support it.
         Insn::SendWithoutBlockDirect { cd, state, args, .. } if args.len() + 1 > C_ARG_OPNDS.len() => // +1 for self
             gen_send_without_block(jit, asm, *cd, &function.frame_state(*state)),
@@ -1995,6 +1998,24 @@ fn aligned_stack_bytes(num_slots: usize) -> usize {
     num_slots * SIZEOF_VALUE
 }
 
+fn update_send_fallback_counter(asm: &mut Assembler, def_type: u32) {
+    match def_type {
+        VM_METHOD_TYPE_ISEQ => { gen_incr_counter(asm, Counter::send_fallback_iseq); }
+        VM_METHOD_TYPE_CFUNC => { gen_incr_counter(asm, Counter::send_fallback_cfunc); }
+        VM_METHOD_TYPE_ATTRSET => { gen_incr_counter(asm, Counter::send_fallback_attrset); }
+        VM_METHOD_TYPE_IVAR => { gen_incr_counter(asm, Counter::send_fallback_ivar); }
+        VM_METHOD_TYPE_BMETHOD => { gen_incr_counter(asm, Counter::send_fallback_bmethod); }
+        VM_METHOD_TYPE_ZSUPER => { gen_incr_counter(asm, Counter::send_fallback_zsuper); }
+        VM_METHOD_TYPE_ALIAS => { gen_incr_counter(asm, Counter::send_fallback_alias); }
+        VM_METHOD_TYPE_UNDEF => { gen_incr_counter(asm, Counter::send_fallback_undef); }
+        VM_METHOD_TYPE_NOTIMPLEMENTED => { gen_incr_counter(asm, Counter::send_fallback_not_implemented); }
+        VM_METHOD_TYPE_OPTIMIZED => { gen_incr_counter(asm, Counter::send_fallback_optimized); }
+        VM_METHOD_TYPE_MISSING => { gen_incr_counter(asm, Counter::send_fallback_missing); }
+        VM_METHOD_TYPE_REFINED => { gen_incr_counter(asm, Counter::send_fallback_refined); }
+        _ => unreachable!("unknown method type: {def_type}"),
+    }
+}
+
 impl Assembler {
     /// Make a C call while marking the start and end positions for IseqCall
     fn ccall_with_iseq_call(&mut self, fptr: *const u8, opnds: Vec<Opnd>, iseq_call: &Rc<RefCell<IseqCall>>) -> Opnd {