1 files changed, 14 insertions, 0 deletions

diff --git a/crates/ra_prof/src/lib.rs b/crates/ra_prof/src/lib.rs
index 1e8d780ab..6f7918745 100644
--- a/crates/ra_prof/src/lib.rs
+++ b/crates/ra_prof/src/lib.rs
@@ -1,3 +1,5 @@
+mod memory_usage;
+
 use std::{
     cell::RefCell,
     time::{Duration, Instant},
@@ -11,6 +13,14 @@ use std::{
 use once_cell::sync::Lazy;
 use itertools::Itertools;
 
+pub use crate::memory_usage::{MemoryUsage, Bytes};
+
+// We use jemalloc mainly to get heap usage statistics, actual performance
+// difference is not measures.
+#[cfg(feature = "jemalloc")]
+#[global_allocator]
+static ALLOC: jemallocator::Jemalloc = jemallocator::Jemalloc;
+
 /// Set profiling filter. It specifies descriptions allowed to profile.
 /// This is helpful when call stack has too many nested profiling scopes.
 /// Additionally filter can specify maximum depth of profiling scopes nesting.
@@ -288,6 +298,10 @@ impl Drop for CpuProfiler {
     }
 }
 
+pub fn memory_usage() -> MemoryUsage {
+    MemoryUsage::current()
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;