Using MiMallocator for better performance (#8534)

Author: eytan-starkware

Cargo.lock

@@ -31,6 +31,10 @@ opt-level = 3
 opt-level = 3
 [profile.ci-dev.package."vector-map"]
 opt-level = 3
+[profile.ci-dev.package."mimalloc"]
+opt-level = 3
+[profile.ci-dev.package."libmimalloc-sys"]
+opt-level = 3
 
 [workspace]
 resolver = "2"
@@ -123,6 +127,7 @@ itertools = { version = "0.14.0", default-features = false }
 keccak = "0.1.5"
 lalrpop-util = { version = "0.22.2", features = ["lexer"] }
 log = "0.4.27"
+mimalloc = { version = "0.1.48" }
 num-bigint = { version = "0.4.6", default-features = false }
 num-integer = "0.1.46"
 num-traits = { version = "0.2.19", default-features = false }

Cargo.toml

@@ -10,8 +10,12 @@ description = "Cairo compiler executable for the Cairo programming language"
 anyhow.workspace = true
 clap.workspace = true
 log.workspace = true
+mimalloc = { workspace = true, optional = true }
 tracing.workspace = true
 
 cairo-lang-compiler = { path = "../../cairo-lang-compiler", version = "~2.13.0" }
 cairo-lang-lowering = { path = "../../cairo-lang-lowering", version = "~2.13.0" }
 cairo-lang-utils = { path = "../../cairo-lang-utils", version = "~2.13.0", features = ["tracing"] }
+
+[features]
+mimalloc = ["dep:mimalloc"]

crates/bin/cairo-compile/Cargo.toml

@@ -7,6 +7,10 @@ use cairo_lang_compiler::{CompilerConfig, compile_cairo_project_at_path};
 use cairo_lang_utils::logging::init_logging;
 use clap::Parser;
 
+#[cfg(feature = "mimalloc")]
+#[global_allocator]
+static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
 /// Options for the `inlining-strategy` arguments.
 #[derive(Copy, Clone, Debug, Default, PartialEq, Eq, clap::ValueEnum)]
 pub enum InliningStrategy {

crates/bin/cairo-compile/src/main.rs

@@ -19,6 +19,11 @@ cairo-lang-runner = { path = "../../cairo-lang-runner", version = "~2.13.0" }
 cairo-lang-sierra-generator = { path = "../../cairo-lang-sierra-generator", version = "~2.13.0" }
 cairo-vm = { workspace = true, features = ["clap"] }
 clap.workspace = true
+mimalloc = { workspace = true, features = ["secure"], optional = true }
 num-bigint.workspace = true
 salsa.workspace = true
 serde_json.workspace = true
+
+[features]
+
+mimalloc = ["dep:mimalloc"]

crates/bin/cairo-execute/Cargo.toml

@@ -30,6 +30,10 @@ use clap::Parser;
 use num_bigint::BigInt;
 use salsa::Database;
 
+#[cfg(feature = "mimalloc")]
+#[global_allocator]
+static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
 /// Compiles a Cairo project and runs a function marked `#[executable]`.
 /// Exits with 1 if the compilation or run fails, otherwise 0.
 #[derive(Parser, Debug)]

crates/bin/cairo-execute/src/main.rs

@@ -11,7 +11,11 @@ clap.workspace = true
 colored.workspace = true
 ignore.workspace = true
 log.workspace = true
+mimalloc = { workspace = true, optional = true }
 tracing.workspace = true
 
 cairo-lang-formatter = { path = "../../cairo-lang-formatter", version = "~2.13.0" }
 cairo-lang-utils = { path = "../../cairo-lang-utils", version = "~2.13.0", features = ["tracing"] }
+
+[features]
+mimalloc = ["dep:mimalloc"]

crates/bin/cairo-format/Cargo.toml

@@ -11,6 +11,10 @@ use ignore::WalkState::Continue;
 use ignore::{DirEntry, Error, ParallelVisitor, ParallelVisitorBuilder, WalkState};
 use log::warn;
 
+#[cfg(feature = "mimalloc")]
+#[global_allocator]
+static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
 /// Outputs a string to stderr if the verbose flag is true.
 fn eprintln_if_verbose(s: &str, verbose: bool) {
     if verbose {

crates/bin/cairo-format/src/main.rs

@@ -9,10 +9,14 @@ description = "Runner executable for the Cairo programming language"
 [dependencies]
 anyhow.workspace = true
 clap.workspace = true
+mimalloc = { workspace = true, optional = true }
 
 cairo-lang-compiler = { path = "../../cairo-lang-compiler", version = "~2.13.0" }
 cairo-lang-diagnostics = { path = "../../cairo-lang-diagnostics", version = "~2.13.0" }
 cairo-lang-filesystem = { path = "../../cairo-lang-filesystem", version = "~2.13.0" }
 cairo-lang-runner = { path = "../../cairo-lang-runner", version = "~2.13.0" }
 cairo-lang-sierra-generator = { path = "../../cairo-lang-sierra-generator", version = "~2.13.0" }
 cairo-lang-starknet = { path = "../../cairo-lang-starknet", version = "~2.13.0" }
+
+[features]
+mimalloc = ["dep:mimalloc"]

crates/bin/cairo-run/Cargo.toml

@@ -18,6 +18,10 @@ use cairo_lang_sierra_generator::replace_ids::{DebugReplacer, SierraIdReplacer};
 use cairo_lang_starknet::contract::{find_contracts, get_contracts_info};
 use clap::Parser;
 
+#[cfg(feature = "mimalloc")]
+#[global_allocator]
+static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
 /// Compiles a Cairo project and runs the function `main`.
 /// Exits with 1 if the compilation or run fails, otherwise 0.
 #[derive(Parser, Debug)]