1 // Copyright (C) 2020 Alibaba Cloud. All rights reserved.
2 //
3 // SPDX-License-Identifier: Apache-2.0 OR BSD-3-Clause
4
5 pub use criterion::{black_box, Criterion};
6 use vm_memory::volatile_memory::VolatileMemory;
7 use vm_memory::VolatileSlice;
8
benchmark_for_volatile(c: &mut Criterion)9 pub fn benchmark_for_volatile(c: &mut Criterion) {
10 let mut a = [0xa5u8; 1024];
11 let vslice = VolatileSlice::from(&mut a[..]);
12 let v_ref8 = vslice.get_slice(0, vslice.len()).unwrap();
13 let mut d8 = [0u8; 1024];
14
15 // Check performance for read operations.
16 c.bench_function("VolatileSlice::copy_to_u8", |b| {
17 b.iter(|| v_ref8.copy_to(black_box(&mut d8[..])))
18 });
19
20 let v_ref16 = vslice.get_slice(0, vslice.len() / 2).unwrap();
21 let mut d16 = [0u16; 512];
22
23 c.bench_function("VolatileSlice::copy_to_u16", |b| {
24 b.iter(|| v_ref16.copy_to(black_box(&mut d16[..])))
25 });
26 benchmark_volatile_copy_to_volatile_slice(c);
27
28 // Check performance for write operations.
29 c.bench_function("VolatileSlice::copy_from_u8", |b| {
30 b.iter(|| v_ref8.copy_from(black_box(&d8[..])))
31 });
32 c.bench_function("VolatileSlice::copy_from_u16", |b| {
33 b.iter(|| v_ref16.copy_from(black_box(&d16[..])))
34 });
35 }
36
benchmark_volatile_copy_to_volatile_slice(c: &mut Criterion)37 fn benchmark_volatile_copy_to_volatile_slice(c: &mut Criterion) {
38 let mut a = [0xa5u8; 10240];
39 let vslice = VolatileSlice::from(&mut a[..]);
40 let a_slice = vslice.get_slice(0, vslice.len()).unwrap();
41 let mut d = [0u8; 10240];
42 let vslice2 = VolatileSlice::from(&mut d[..]);
43 let d_slice = vslice2.get_slice(0, vslice2.len()).unwrap();
44
45 c.bench_function("VolatileSlice::copy_to_volatile_slice", |b| {
46 b.iter(|| black_box(a_slice).copy_to_volatile_slice(d_slice))
47 });
48 }
49