Benchmark AllocProfiler internals

nvzqz · Dec 11, 2023 · c605ab7 · c605ab7
1 parent 43d2ddc
commit c605ab7
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/internal_benches/benches/internals.rs b/internal_benches/benches/internals.rs
@@ -1,3 +1,8 @@
+use divan::AllocProfiler;
+
+#[global_allocator]
+static GLOBAL_ALLOC: AllocProfiler = AllocProfiler::system();
+
 fn main() {
     divan::main();
 }
diff --git a/src/alloc.rs b/src/alloc.rs
@@ -562,3 +562,18 @@ impl<T> AllocOpMap<T> {
         &self.values[op as usize]
     }
 }
+
+#[cfg(feature = "internal_benches")]
+mod thread_info {
+    use super::*;
+
+    #[crate::bench(crate = crate)]
+    fn current() -> &'static ThreadAllocInfo {
+        AllocProfiler::system().current_thread_info()
+    }
+
+    #[crate::bench(crate = crate)]
+    fn try_current() -> Option<&'static ThreadAllocInfo> {
+        CURRENT_THREAD_INFO.get()
+    }
+}