feat(hydro_lang): use mimalloc when compiling for an external host

Author: shadaj

Cargo.lock

@@ -48,6 +48,7 @@ build = ["dep:dfir_lang", "dep:backtrace", "dep:ctor"]
 trybuild = ["build", "dep:toml", "dep:prettyplease", "dep:stageleft_tool", "dep:trybuild-internals-api", "dep:sha2"]
 runtime_measure = ["dep:chrono", "dep:procfs"]
 runtime_support = ["dep:dfir_rs", "dfir_rs/deploy_integration"]
+runtime_mimalloc = ["dep:mimalloc"]
 dfir_context = ["dep:dfir_rs"]
 
 [package.metadata.docs.rs]
@@ -65,6 +66,7 @@ dfir_rs = { path = "../dfir_rs", version = "^0.14.0", default-features = false,
 dfir_lang = { path = "../dfir_lang", version = "^0.14.0", optional = true }
 futures = "0.3.0"
 match_box = "0.0.2"
+mimalloc = { version = "0.1", optional = true }
 nameof = { version = "1.0.0", optional = true }
 prettyplease = { version = "0.2.0", features = ["verbatim"], optional = true }
 proc-macro-crate = "3.3"

hydro_lang/Cargo.toml

@@ -18,7 +18,7 @@ use trybuild_internals_api::{Runner, dependencies, features, path};
 #[cfg(feature = "deploy")]
 use super::rewriters::UseTestModeStaged;
 
-pub const HYDRO_RUNTIME_FEATURES: &[&str] = &["deploy_integration", "runtime_measure"];
+pub const HYDRO_RUNTIME_FEATURES: &[&str] = &["deploy_integration", "runtime_measure", "runtime_mimalloc"];
 
 pub(crate) static IS_TEST: std::sync::atomic::AtomicBool =
     std::sync::atomic::AtomicBool::new(false);
@@ -181,6 +181,9 @@ pub fn compile_graph_trybuild(
         use hydro_lang::runtime_support::dfir_rs as __root_dfir_rs;
         pub use #trybuild_crate_name_ident::__staged;
 
+        #[global_allocator]
+        static GLOBAL: hydro_lang::runtime_support::mimalloc::MiMalloc = hydro_lang::runtime_support::mimalloc::MiMalloc;
+
         #[allow(unused)]
         fn __hydro_runtime<'a>(__hydro_lang_trybuild_cli: &'a hydro_lang::runtime_support::dfir_rs::util::deploy::DeployPorts<hydro_lang::__staged::deploy::deploy_runtime::HydroMeta>) -> hydro_lang::runtime_support::dfir_rs::scheduled::graph::Dfir<'a> {
             #(#extra_stmts)*

hydro_lang/src/compile/trybuild/generate.rs

@@ -15,7 +15,7 @@ use hydro_deploy::custom_service::CustomClientPort;
 use hydro_deploy::rust_crate::RustCrateService;
 use hydro_deploy::rust_crate::ports::{DemuxSink, RustCrateSink, RustCrateSource, TaggedSource};
 use hydro_deploy::rust_crate::tracing_options::TracingOptions;
-use hydro_deploy::{CustomService, Deployment, Host, RustCrate, TracingResults};
+use hydro_deploy::{CustomService, Deployment, Host, HostTargetType, RustCrate, TracingResults};
 use hydro_deploy_integration::{ConnectedSink, ConnectedSource};
 use nameof::name_of;
 use proc_macro2::Span;
@@ -1043,6 +1043,7 @@ fn create_trybuild_service(
     features: &Option<Vec<String>>,
     bin_name: &str,
 ) -> RustCrate {
+    let target_type = trybuild.host.target_type();
     let mut ret = RustCrate::new(dir, trybuild.host)
         .target_dir(target_dir)
         .example(bin_name)
@@ -1084,6 +1085,10 @@ fn create_trybuild_service(
             .chain(trybuild.features),
     );
 
+    if target_type != HostTargetType::Local {
+        ret = ret.features(["hydro___feature_runtime_mimalloc".to_string()]);
+    }
+
     for (key, value) in trybuild.build_envs {
         ret = ret.build_env(key, value);
     }

hydro_lang/src/deploy/deploy_graph.rs

@@ -22,6 +22,9 @@ pub mod runtime_support {
     pub use colored;
     pub use {bincode, dfir_rs, stageleft, tokio};
     pub mod resource_measurement;
+
+    #[cfg(feature = "runtime_mimalloc")]
+    pub use mimalloc;
 }
 
 #[doc(hidden)]