1 // Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
2 //
3 // SPDX-License-Identifier: Apache-2.0 OR BSD-3-Clause
4
5 extern crate criterion;
6
7 pub use criterion::{black_box, criterion_group, criterion_main, Criterion};
8 #[cfg(feature = "backend-mmap")]
9 use vm_memory::{GuestAddress, GuestMemoryMmap};
10
11 mod guest_memory;
12 mod mmap;
13 mod volatile;
14
15 use volatile::benchmark_for_volatile;
16
17 #[cfg(feature = "backend-mmap")]
18 // Use this function with caution. It does not check against overflows
19 // and `GuestMemoryMmap::from_ranges` errors.
create_guest_memory_mmap(size: usize, count: u64) -> GuestMemoryMmap<()>20 fn create_guest_memory_mmap(size: usize, count: u64) -> GuestMemoryMmap<()> {
21 let mut regions: Vec<(GuestAddress, usize)> = Vec::new();
22 for i in 0..count {
23 regions.push((GuestAddress(i * size as u64), size));
24 }
25
26 GuestMemoryMmap::from_ranges(regions.as_slice()).unwrap()
27 }
28
criterion_benchmark(_c: &mut Criterion)29 pub fn criterion_benchmark(_c: &mut Criterion) {
30 #[cfg(feature = "backend-mmap")]
31 mmap::benchmark_for_mmap(_c);
32 }
33
benchmark_guest_memory(_c: &mut Criterion)34 pub fn benchmark_guest_memory(_c: &mut Criterion) {
35 #[cfg(feature = "backend-mmap")]
36 guest_memory::benchmark_for_guest_memory(_c)
37 }
38
39 criterion_group! {
40 name = benches;
41 config = Criterion::default().sample_size(200).measurement_time(std::time::Duration::from_secs(50));
42 targets = criterion_benchmark, benchmark_guest_memory, benchmark_for_volatile
43 }
44
45 criterion_main! {
46 benches,
47 }
48