Add instrumentation::setprofile_all_threads

PyO3 · Jun 23, 2024 · 3ae8ef4 · 3ae8ef4
1 parent 2e184e3
commit 3ae8ef4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/instrumentation.rs b/src/instrumentation.rs
@@ -62,6 +62,12 @@ pub fn setprofile<P: Profiler>(profiler: Bound<'_, P>) {
     unsafe { ffi::PyEval_SetProfile(Some(profile_callback::<P>), profiler.into_ptr()) };
 }
 
+/// Register a custom Profiler with the Python interpreter.
+#[cfg(Py_3_12)]
+pub fn setprofile_all_threads<P: Profiler>(profiler: Bound<'_, P>) {
+    unsafe { ffi::PyEval_SetProfileAllThreads(Some(profile_callback::<P>), profiler.into_ptr()) };
+}
+
 extern "C" fn profile_callback<P>(
     obj: *mut ffi::PyObject,
     frame: *mut ffi::PyFrameObject,