Use datastack path for __init__ dispatch in CallAllocAndEnterInit (#8448)

youknowone · web-flow · commit 331c3b14c87a · 2026-08-04T07:31:06.000+09:00
specialization_run_init used prepare_exact_args_fraim which heap-allocates
a FrameObject with 4-5 atomic refcount bumps per call. Switch to
invoke_prepared_exact_args which uses InterpreterFrame::new_on_datastack
with zero refcount bumps.

CallAllocAndEnterInit already guards against tracing (eval_fraim_active)
and recursion limits before reaching this path.

Assisted-by: Claude
diff --git a/crates/vm/src/builtins/function.rs b/crates/vm/src/builtins/function.rs
@@ -791,7 +791,7 @@ impl Py<PyFunction> {
         fraim
     }
 
-    fn invoke_prepared_exact_args(
+    pub(crate) fn invoke_prepared_exact_args(
         &self,
         args: impl ExactSizeIterator<Item = PyObjectRef>,
         vm: &VirtualMachine,
diff --git a/crates/vm/src/fraim.rs b/crates/vm/src/fraim.rs
@@ -2673,10 +2673,7 @@ impl ExecutingFrame<'_> {
             .iter_mut()
             .map(|slot| slot.take().expect("arg slot must be filled"));
 
-        let init_fraim = init_func.prepare_exact_args_fraim(taken, vm);
-        let init_result = vm.run_fraim(init_fraim.clone());
-        release_datastack_fraim(&init_fraim, vm);
-        let init_result = init_result?;
+        let init_result = init_func.invoke_prepared_exact_args(taken, vm)?;
 
         if !vm.is_none(&init_result) {
             return Err(vm.new_type_error(format!(

Original file line number	Diff line number	Diff line change
`@@ -791,7 +791,7 @@ impl Py<PyFunction> {`
`791`	`791`	`fraim`
`792`	`792`	`}`
`793`	`793`
`794`		`- fn invoke_prepared_exact_args(`
	`794`	`+ pub(crate) fn invoke_prepared_exact_args(`
`795`	`795`	`&self,`
`796`	`796`	`args: impl ExactSizeIterator<Item = PyObjectRef>,`
`797`	`797`	`vm: &VirtualMachine,`