github/powdr
1
1
Fork 0
mirror of https://github.com/powdr-labs/powdr.git synced 2026-04-20 03:03:25 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #872 from powdr-labs/add-benchmark

Browse Source 
Add `many_chunks` example to benchmarks
This commit is contained in:
Leo
2024-01-05 19:46:39 +00:00
committed by GitHub
parent d897805afd b7d89fbbef
commit 4d9406def8
1 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
compiler/benches/executor_benchmark.rs
View File

@@ -5,21 +5,29 @@ use criterion::{criterion_group, criterion_main, Criterion};
use mktemp::Temp;
use number::{FieldElement, GoldilocksField};
use riscv::{compile_rust_crate_to_riscv_asm, compiler};
use riscv::continuations::bootloader::default_input;
use riscv::{compile_rust_crate_to_riscv_asm, compile_rust_to_riscv_asm, compiler};
use riscv::CoProcessors;
type T = GoldilocksField;
fn run_witgen<T: FieldElement>(analyzed: &Analyzed<T>, constants: Vec<(String, Vec<T>)>) {
fn run_witgen<T: FieldElement>(
analyzed: &Analyzed<T>,
constants: Vec<(String, Vec<T>)>,
external_witness_values: Vec<(String, Vec<T>)>,
) {
let query_callback = inputs_to_query_callback(vec![]);
executor::witgen::WitnessGenerator::new(analyzed, &constants, query_callback).generate();
executor::witgen::WitnessGenerator::new(analyzed, &constants, query_callback)
.with_external_witness_values(external_witness_values)
.generate();
}
fn criterion_benchmark(c: &mut Criterion) {
let mut group = c.benchmark_group("keccak-executor-benchmark");
let mut group = c.benchmark_group("executor-benchmark");
group.sample_size(10);
// Keccak
let tmp_dir = Temp::new_dir().unwrap();
let riscv_asm_files =
compile_rust_crate_to_riscv_asm("../riscv/tests/riscv_data/keccak/Cargo.toml", &tmp_dir);
@@ -34,6 +42,26 @@ fn criterion_benchmark(c: &mut Criterion) {
run_witgen(
&pil_with_constants.pil,
pil_with_constants.fixed_cols.clone(),
vec![],
)
})
});
// The first chunk of `many_chunks`, with Poseidon co-processor & bootloader
let riscv_asm_files =
compile_rust_to_riscv_asm("../riscv/tests/riscv_data/many_chunks.rs", &tmp_dir);
let contents = compiler::compile(riscv_asm_files, &CoProcessors::base().with_poseidon(), true);
let pil_with_constants = Pipeline::<T>::default()
.from_asm_string(contents, None)
.optimized_pil_with_constants()
.unwrap();
group.bench_function("many_chunks_chunk_0", |b| {
b.iter(|| {
run_witgen(
&pil_with_constants.pil,
pil_with_constants.fixed_cols.clone(),
vec![("main.bootloader_input_value".to_string(), default_input())],
)
})
});