59 lines
2.3 KiB
Rust
59 lines
2.3 KiB
Rust
// Configure jemalloc as the `global_allocator` when configured. This is
|
|
// so that we use the sized deallocation apis jemalloc provides
|
|
// (namely `sdallocx`).
|
|
//
|
|
// The symbol overrides documented below are also performed so that we can
|
|
// ensure that we use a consistent allocator across the rustc <-> llvm boundary
|
|
#[cfg(feature = "jemalloc")]
|
|
#[global_allocator]
|
|
static ALLOC: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
|
|
|
|
#[cfg(feature = "tikv-jemalloc-sys")]
|
|
use tikv_jemalloc_sys as jemalloc_sys;
|
|
|
|
fn main() {
|
|
// Pull in jemalloc when enabled.
|
|
//
|
|
// Note that we're pulling in a static copy of jemalloc which means that to
|
|
// pull it in we need to actually reference its symbols for it to get
|
|
// linked. The two crates we link to here, std and rustc_driver, are both
|
|
// dynamic libraries. That means to pull in jemalloc we actually need to
|
|
// reference allocation symbols one way or another (as this file is the only
|
|
// object code in the rustc executable).
|
|
#[cfg(feature = "tikv-jemalloc-sys")]
|
|
{
|
|
use std::os::raw::{c_int, c_void};
|
|
|
|
#[used]
|
|
static _F1: unsafe extern "C" fn(usize, usize) -> *mut c_void = jemalloc_sys::calloc;
|
|
#[used]
|
|
static _F2: unsafe extern "C" fn(*mut *mut c_void, usize, usize) -> c_int =
|
|
jemalloc_sys::posix_memalign;
|
|
#[used]
|
|
static _F3: unsafe extern "C" fn(usize, usize) -> *mut c_void = jemalloc_sys::aligned_alloc;
|
|
#[used]
|
|
static _F4: unsafe extern "C" fn(usize) -> *mut c_void = jemalloc_sys::malloc;
|
|
#[used]
|
|
static _F5: unsafe extern "C" fn(*mut c_void, usize) -> *mut c_void = jemalloc_sys::realloc;
|
|
#[used]
|
|
static _F6: unsafe extern "C" fn(*mut c_void) = jemalloc_sys::free;
|
|
|
|
// On OSX, jemalloc doesn't directly override malloc/free, but instead
|
|
// registers itself with the allocator's zone APIs in a ctor. However,
|
|
// the linker doesn't seem to consider ctors as "used" when statically
|
|
// linking, so we need to explicitly depend on the function.
|
|
#[cfg(target_os = "macos")]
|
|
{
|
|
extern "C" {
|
|
fn _rjem_je_zone_register();
|
|
}
|
|
|
|
#[used]
|
|
static _F7: unsafe extern "C" fn() = _rjem_je_zone_register;
|
|
}
|
|
}
|
|
|
|
rustc_driver::set_sigpipe_handler();
|
|
rustc_driver::main()
|
|
}
|