Use mimalloc as the allocator on x86_64-pc-windows-msvc

Author: Zoxc

Cargo.lock

@@ -756,7 +756,7 @@ checksum = "af491d569909a7e4dee0ad7db7f5341fef5c614d5b8ec8cf765732aba3cff681"
 dependencies = [
  "serde",
  "termcolor",
- "unicode-width 0.1.14",
+ "unicode-width 0.2.2",
 ]
 
 [[package]]
@@ -977,6 +977,12 @@ dependencies = [
  "windows-sys 0.61.2",
 ]
 
+[[package]]
+name = "cty"
+version = "0.2.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b365fabc795046672053e29c954733ec3b05e4be654ab130fe8f1f94d7051f35"
+
 [[package]]
 name = "curl"
 version = "0.4.49"
@@ -2216,6 +2222,16 @@ version = "0.2.15"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "f9fbbcab51052fe104eb5e5d351cf728d30a5be1fe14d9be8a3b097481fb97de"
 
+[[package]]
+name = "libmimalloc-sys"
+version = "0.1.47"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2d1eacfa31c33ec25e873c136ba5669f00f9866d0688bea7be4d3f7e43067df6"
+dependencies = [
+ "cc",
+ "cty",
+]
+
 [[package]]
 name = "libredox"
 version = "0.1.10"
@@ -2399,6 +2415,15 @@ dependencies = [
  "libc",
 ]
 
+[[package]]
+name = "mimalloc"
+version = "0.1.50"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b3627c4272df786b9260cabaa46aec1d59c93ede723d4c3ef646c503816b0640"
+dependencies = [
+ "libmimalloc-sys",
+]
+
 [[package]]
 name = "minifier"
 version = "0.3.6"
@@ -3808,6 +3833,8 @@ dependencies = [
  "ctrlc",
  "jiff",
  "libc",
+ "libmimalloc-sys",
+ "mimalloc",
  "rustc_abi",
  "rustc_ast",
  "rustc_ast_pretty",

compiler/rustc/Cargo.toml

@@ -40,6 +40,7 @@ llvm = ['rustc_driver_impl/llvm']
 llvm_enzyme = ['rustc_driver_impl/llvm_enzyme']
 llvm_offload = ['rustc_driver_impl/llvm_offload']
 max_level_info = ['rustc_driver_impl/max_level_info']
+mimalloc = ['rustc_driver_impl/mimalloc']
 rustc_randomized_layouts = ['rustc_driver_impl/rustc_randomized_layouts']
 # tidy-alphabetical-end
 

compiler/rustc_driver_impl/Cargo.toml

@@ -55,13 +55,23 @@ features = [
 ctrlc = "3.4.4"
 # tidy-alphabetical-end
 
+[target.'cfg(target_os = "windows")'.dependencies.mimalloc]
+version = "0.1.50"
+optional = true
+
+[target.'cfg(target_os = "windows")'.dependencies.libmimalloc-sys]
+version = "0.1.47"
+optional = true
+features = ["extended"]
+
 [features]
 # tidy-alphabetical-start
 check_only = ['rustc_interface/check_only']
 llvm = ['rustc_interface/llvm']
 llvm_enzyme = ['rustc_interface/llvm_enzyme']
 llvm_offload = ['rustc_interface/llvm_offload']
 max_level_info = ['rustc_log/max_level_info']
+mimalloc = ['dep:mimalloc', 'dep:libmimalloc-sys']
 rustc_randomized_layouts = [
     'rustc_index/rustc_randomized_layouts',
     'rustc_middle/rustc_randomized_layouts'

compiler/rustc_driver_impl/src/alloc.rs

@@ -0,0 +1,54 @@
+#[cfg(all(target_os = "windows", feature = "mimalloc"))]
+#[global_allocator]
+static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
+#[cfg(all(target_os = "windows", feature = "mimalloc"))]
+pub mod fns {
+    use std::os::raw::{c_char, c_int, c_void};
+
+    #[unsafe(no_mangle)]
+    pub unsafe extern "C" fn calloc(items: usize, size: usize) -> *mut c_void {
+        unsafe { libmimalloc_sys::mi_calloc(items, size) }
+    }
+
+    #[unsafe(no_mangle)]
+    pub unsafe extern "C" fn posix_memalign(
+        ptr: *mut *mut c_void,
+        align: usize,
+        size: usize,
+    ) -> c_int {
+        unsafe { libmimalloc_sys::mi_posix_memalign(ptr, align, size) }
+    }
+
+    #[unsafe(no_mangle)]
+    pub unsafe extern "C" fn strndup(s: *const c_char, n: usize) -> *mut c_char {
+        unsafe { libmimalloc_sys::mi_strndup(s, n) }
+    }
+
+    #[unsafe(no_mangle)]
+    pub unsafe extern "C" fn strdup(s: *const c_char) -> *mut c_char {
+        unsafe { libmimalloc_sys::mi_strdup(s) }
+    }
+
+    #[unsafe(no_mangle)]
+    pub unsafe extern "C" fn aligned_alloc(align: usize, size: usize) -> *mut c_void {
+        unsafe { libmimalloc_sys::mi_aligned_alloc(align, size) }
+    }
+
+    #[unsafe(no_mangle)]
+    pub unsafe extern "C" fn malloc(size: usize) -> *mut c_void {
+        unsafe { libmimalloc_sys::mi_malloc(size) }
+    }
+
+    #[unsafe(no_mangle)]
+    pub unsafe extern "C" fn realloc(ptr: *mut c_void, size: usize) -> *mut c_void {
+        unsafe { libmimalloc_sys::mi_realloc(ptr, size) }
+    }
+
+    #[unsafe(no_mangle)]
+    pub unsafe extern "C" fn free(ptr: *mut c_void) {
+        unsafe {
+            libmimalloc_sys::mi_free(ptr);
+        }
+    }
+}

compiler/rustc_driver_impl/src/lib.rs

@@ -63,6 +63,8 @@ use rustc_target::json::ToJson;
 use rustc_target::spec::{Target, TargetTuple};
 use tracing::trace;
 
+pub mod alloc;
+
 #[allow(unused_macros)]
 macro do_not_use_print($($t:tt)*) {
     std::compile_error!(

src/bootstrap/src/core/config/mod.rs

@@ -254,12 +254,14 @@ impl<'de> Deserialize<'de> for CompilerBuiltins {
 #[derive(Copy, Clone, Debug, PartialEq, Eq)]
 pub enum OverrideAllocator {
     Jemalloc,
+    Mimalloc,
 }
 
 impl OverrideAllocator {
     pub fn feature_name(self) -> &'static str {
         match self {
             OverrideAllocator::Jemalloc => "jemalloc",
+            OverrideAllocator::Mimalloc => "mimalloc",
         }
     }
 }
@@ -272,7 +274,8 @@ impl<'de> Deserialize<'de> for OverrideAllocator {
         let name = String::deserialize(deserializer)?;
         match name.as_str() {
             "jemalloc" => Ok(Self::Jemalloc),
-            other => Err(serde::de::Error::unknown_variant(other, &["jemalloc"])),
+            "mimalloc" => Ok(Self::Mimalloc),
+            other => Err(serde::de::Error::unknown_variant(other, &["jemalloc", "mimalloc"])),
         }
     }
 }

src/ci/github-actions/jobs.yml

@@ -659,6 +659,7 @@ auto:
         --target=x86_64-pc-windows-msvc
         --enable-full-tools
         --enable-profiler
+        --set rust.override-allocator=mimalloc
         --set rust.codegen-units=1
       SCRIPT: python x.py build --set rust.debug=true opt-dist && PGO_HOST=x86_64-pc-windows-msvc ./build/x86_64-pc-windows-msvc/stage1-tools-bin/opt-dist windows-ci -- python x.py dist bootstrap --include-default-paths
       DIST_REQUIRE_ALL_TOOLS: 1

src/tools/clippy/Cargo.toml

@@ -57,6 +57,7 @@ rustc_tools_util = { path = "rustc_tools_util", version = "0.4.2" }
 integration = ["dep:tempfile"]
 internal = ["dep:clippy_lints_internal", "dep:tempfile"]
 jemalloc = []
+mialloc = []
 
 [package.metadata.rust-analyzer]
 # This package uses #[feature(rustc_private)]

src/tools/miri/Cargo.toml

@@ -69,6 +69,7 @@ expensive-consistency-checks = ["stack-cache"]
 tracing = ["serde_json"]
 native-lib = ["dep:libffi", "dep:libloading", "dep:capstone", "dep:ipc-channel", "dep:nix", "dep:serde"]
 jemalloc = []
+mialloc = []
 check_only = ["libffi?/check_only", "capstone?/check_only", "genmc-sys?/check_only"]
 
 [lints.rust.unexpected_cfgs]

src/tools/rustdoc/Cargo.toml

@@ -15,3 +15,4 @@ rustdoc = { path = "../../librustdoc" }
 
 [features]
 jemalloc = ['rustdoc/jemalloc']
+mialloc = []