oxc/tasks/benchmark/src/main.rs

#![cfg(not(miri))] // Miri does not support custom allocators

#[cfg(not(target_env = "msvc"))]
#[global_allocator]
static GLOBAL: jemallocator::Jemalloc = jemallocator::Jemalloc;

#[cfg(target_os = "windows")]
#[global_allocator]
static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;

// See: `https://rust-lang.github.io/rfcs/2360-bench-black-box.html`
use std::hint::black_box;
use std::time::Duration;

use criterion::{BenchmarkId, Criterion, Throughput};
use oxc_allocator::Allocator;
use oxc_ast::SourceType;
use oxc_parser::Parser;
use oxc_semantic::SemanticBuilder;
use oxc_tasks_common::TestFiles;
use pico_args::Arguments;

/// # Errors
/// # Panics
pub fn main() -> Result<(), String> {
    let files = TestFiles::new();
    let mut args = Arguments::from_env();

    let baseline: Option<String> = args.opt_value_from_str("--save-baseline").unwrap();
    let measurement_time = Duration::new(/* seconds */ 10, 0);
    let mut criterion = Criterion::default().without_plots().measurement_time(measurement_time);
    if let Some(ref baseline) = baseline {
        criterion = criterion.save_baseline(baseline.to_string());
    }

    // Check files
    for file in files.files() {
        let allocator = Allocator::default();
        let ret =
            Parser::new(&allocator, black_box(&file.source_text), SourceType::default()).parse();
        if !ret.errors.is_empty() {
            println!("{} failed", &file.file_name);
            for error in &ret.errors {
                println!("{error:?}");
            }
            return Err("Parse Failed.".to_string());
        }
    }

    bench_parser(&mut criterion, &files);
    bench_semantic(&mut criterion, &files);
    drop(criterion);

    Ok(())
}

fn bench_parser(criterion: &mut Criterion, files: &TestFiles) {
    let mut group = criterion.benchmark_group("parser");
    for file in files.files() {
        group.throughput(Throughput::Bytes(file.source_text.len() as u64));
        group.bench_with_input(
            BenchmarkId::from_parameter(&file.file_name),
            &file.source_text,
            |b, source_text| {
                let allocator = Allocator::default();
                b.iter(|| {
                    let _drop =
                        Parser::new(&allocator, black_box(source_text), SourceType::default())
                            .parse();
                });
            },
        );
    }
    group.finish();
}

fn bench_semantic(criterion: &mut Criterion, files: &TestFiles) {
    let mut group = criterion.benchmark_group("semantic");
    for file in files.files() {
        group.throughput(Throughput::Bytes(file.source_text.len() as u64));
        group.bench_with_input(
            BenchmarkId::from_parameter(&file.file_name),
            &file.source_text,
            |b, source_text| {
                let allocator = Allocator::default();
                let source_type = SourceType::from_path(&file.file_name).unwrap();
                let ret = Parser::new(&allocator, source_text, source_type).parse();
                let program = allocator.alloc(ret.program);
                b.iter(|| {
                    let _semantic = SemanticBuilder::new(source_text, source_type, &ret.trivias)
                        .build(black_box(program));
                });
            },
        );
    }
    group.finish();
}