perf: Use jemallocator by default for taso (#198)

Also enables thin LTO on release builds. Tested on `barenco_tof_10`, 10s timeout, single threaded, on my laptop. Before / after: <img src="https://github.com/CQCL/tket2/assets/121866228/a3bec2d1-c985-4462-8404-d3c6d3baf4ea" width="49%"/> <img src="https://github.com/CQCL/tket2/assets/121866228/c39e2bfc-b1b3-47a3-ae4f-837870a8b2fb" width="49%"/> --------- Co-authored-by: doug-q <[email protected]>
CQCL · Oct 24, 2023 · 038bb70 · 038bb70
1 parent fb4c668
commit 038bb70
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -8,6 +8,9 @@ bench = false
 name = "tket2"
 path = "src/lib.rs"
 
+[profile.release]
+lto = "thin"
+
 [dependencies]
 lazy_static = "1.4.0"
 cgmath = "0.18.0"

diff --git a/taso-optimiser/Cargo.toml b/taso-optimiser/Cargo.toml
@@ -17,6 +17,9 @@ tracing-subscriber = "0.3.17"
 tracing-appender = "0.2.2"
 peak_alloc = { version = "0.2.0", optional = true }
 
+[target.'cfg(not(target_env = "msvc"))'.dependencies]
+tikv-jemallocator = "0.5"
+
 [features]
-default = ["peak_alloc"]
+default = []
 peak_alloc = ["dep:peak_alloc"]
diff --git a/taso-optimiser/src/main.rs b/taso-optimiser/src/main.rs
@@ -15,11 +15,12 @@ use tket2::optimiser::taso::log::TasoLogger;
 use tket2::optimiser::TasoOptimiser;
 
 #[cfg(feature = "peak_alloc")]
-use peak_alloc::PeakAlloc;
+#[global_allocator]
+static PEAK_ALLOC: peak_alloc::PeakAlloc = peak_alloc::PeakAlloc;
 
-#[cfg(feature = "peak_alloc")]
+#[cfg(all(not(target_env = "msvc"), not(feature = "peak_alloc")))]
 #[global_allocator]
-static PEAK_ALLOC: PeakAlloc = PeakAlloc;
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
 
 /// Optimise circuits using Quartz-generated ECCs.
 ///