d0ab3c8d15
Switch tvix to using jemalloc as the default global allocator on
supported (eg, non-msvc) platforms. This gives a pretty reasonable
performance boost basically for free:
int time: [24.361 µs 24.386 µs 24.418 µs]
change: [-19.355% -18.859% -18.527%] (p = 0.00 < 0.05)
Performance has improved.
merge small attrs time: [37.201 µs 37.328 µs 37.442 µs]
change: [-24.609% -24.266% -23.982%] (p = 0.00 < 0.05)
Performance has improved.
merge large attrs with small attrs
time: [20.030 ms 20.135 ms 20.251 ms]
change: [-4.2811% -3.2549% -2.3807%] (p = 0.00 < 0.05)
Performance has improved.
hello outpath time: [967.91 ms 974.07 ms 983.33 ms]
change: [-5.5008% -4.4552% -3.4099%] (p = 0.00 < 0.05)
Performance has improved.
Change-Id: I6c6e6e3295ec2fca01ea28dc37bcb201cd811767
Reviewed-on: https://cl.tvl.fyi/c/depot/+/10851
Tested-by: BuildkiteCI
Reviewed-by: sterni <sternenseemann@systemli.org>
Autosubmit: aspen <root@gws.fyi>
Reviewed-by: flokli <flokli@flokli.de>
77 lines
2.5 KiB
Rust
77 lines
2.5 KiB
Rust
use criterion::{black_box, criterion_group, criterion_main, Criterion};
|
|
use lazy_static::lazy_static;
|
|
use std::{env, rc::Rc, sync::Arc, time::Duration};
|
|
#[cfg(not(target_env = "msvc"))]
|
|
use tikv_jemallocator::Jemalloc;
|
|
use tvix_build::buildservice::DummyBuildService;
|
|
use tvix_eval::{builtins::impure_builtins, EvalIO};
|
|
use tvix_glue::{
|
|
builtins::{add_derivation_builtins, add_fetcher_builtins, add_import_builtins},
|
|
configure_nix_path,
|
|
tvix_io::TvixIO,
|
|
tvix_store_io::TvixStoreIO,
|
|
};
|
|
use tvix_store::utils::construct_services;
|
|
|
|
#[cfg(not(target_env = "msvc"))]
|
|
#[global_allocator]
|
|
static GLOBAL: Jemalloc = Jemalloc;
|
|
|
|
lazy_static! {
|
|
static ref TOKIO_RUNTIME: tokio::runtime::Runtime = tokio::runtime::Runtime::new().unwrap();
|
|
}
|
|
|
|
fn interpret(code: &str) {
|
|
// TODO: this is a bit annoying.
|
|
// It'd be nice if we could set this up once and then run evaluate() with a
|
|
// piece of code. b/262
|
|
let (blob_service, directory_service, path_info_service, nar_calculation_service) =
|
|
TOKIO_RUNTIME
|
|
.block_on(async { construct_services("memory://", "memory://", "memory://").await })
|
|
.unwrap();
|
|
|
|
// We assemble a complete store in memory.
|
|
let tvix_store_io = Rc::new(TvixStoreIO::new(
|
|
blob_service,
|
|
directory_service,
|
|
path_info_service.into(),
|
|
nar_calculation_service.into(),
|
|
Arc::<DummyBuildService>::default(),
|
|
TOKIO_RUNTIME.handle().clone(),
|
|
));
|
|
|
|
let mut eval = tvix_eval::Evaluation::new(
|
|
Box::new(TvixIO::new(tvix_store_io.clone() as Rc<dyn EvalIO>)) as Box<dyn EvalIO>,
|
|
true,
|
|
);
|
|
|
|
eval.builtins.extend(impure_builtins());
|
|
add_derivation_builtins(&mut eval, Rc::clone(&tvix_store_io));
|
|
add_fetcher_builtins(&mut eval, Rc::clone(&tvix_store_io));
|
|
add_import_builtins(&mut eval, tvix_store_io);
|
|
configure_nix_path(
|
|
&mut eval,
|
|
// The benchmark requires TVIX_BENCH_NIX_PATH to be set, so barf out
|
|
// early, rather than benchmarking tvix returning an error.
|
|
&Some(env::var("TVIX_BENCH_NIX_PATH").expect("TVIX_BENCH_NIX_PATH must be set")),
|
|
);
|
|
|
|
let result = eval.evaluate(code, None);
|
|
|
|
assert!(result.errors.is_empty());
|
|
}
|
|
|
|
fn eval_nixpkgs(c: &mut Criterion) {
|
|
c.bench_function("hello outpath", |b| {
|
|
b.iter(|| {
|
|
interpret(black_box("(import <nixpkgs> {}).hello.outPath"));
|
|
})
|
|
});
|
|
}
|
|
|
|
criterion_group!(
|
|
name = benches;
|
|
config = Criterion::default().measurement_time(Duration::from_secs(30)).sample_size(10);
|
|
targets = eval_nixpkgs
|
|
);
|
|
criterion_main!(benches);
|