summaryrefslogtreecommitdiffstats
path: root/vendor/zerovec/benches/zerovec_serde.rs
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-17 12:18:32 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-17 12:18:32 +0000
commit4547b622d8d29df964fa2914213088b148c498fc (patch)
tree9fc6b25f3c3add6b745be9a2400a6e96140046e9 /vendor/zerovec/benches/zerovec_serde.rs
parentReleasing progress-linux version 1.66.0+dfsg1-1~progress7.99u1. (diff)
downloadrustc-4547b622d8d29df964fa2914213088b148c498fc.tar.xz
rustc-4547b622d8d29df964fa2914213088b148c498fc.zip
Merging upstream version 1.67.1+dfsg1.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'vendor/zerovec/benches/zerovec_serde.rs')
-rw-r--r--vendor/zerovec/benches/zerovec_serde.rs145
1 files changed, 145 insertions, 0 deletions
diff --git a/vendor/zerovec/benches/zerovec_serde.rs b/vendor/zerovec/benches/zerovec_serde.rs
new file mode 100644
index 000000000..3a9bb051a
--- /dev/null
+++ b/vendor/zerovec/benches/zerovec_serde.rs
@@ -0,0 +1,145 @@
+// This file is part of ICU4X. For terms of use, please see the file
+// called LICENSE at the top level of the ICU4X source tree
+// (online at: https://github.com/unicode-org/icu4x/blob/main/LICENSE ).
+
+use criterion::{black_box, criterion_group, criterion_main, Criterion};
+use rand::SeedableRng;
+use rand_distr::{Distribution, LogNormal};
+use rand_pcg::Lcg64Xsh32;
+
+#[path = "../src/samples.rs"]
+mod samples;
+use samples::*;
+
+use zerovec::ZeroVec;
+
+/// Generate a large list of u32s for stress testing.
+#[allow(dead_code)]
+fn random_numbers(count: usize) -> Vec<u32> {
+ // Lcg64Xsh32 is a small, fast PRNG for reproducible benchmarks.
+ // LogNormal(10, 1) generates numbers with mean 36315 and mode 8103, a distribution that, in
+ // spirit, correlates with Unicode properties (many low values and a long tail of high values)
+ let mut rng = Lcg64Xsh32::seed_from_u64(2021);
+ let dist = LogNormal::new(10.0, 1.0).unwrap();
+ (&dist)
+ .sample_iter(&mut rng)
+ .take(count)
+ .map(|f| f as u32)
+ .collect()
+}
+
+fn overview_bench(c: &mut Criterion) {
+ c.bench_function("zerovec_serde/overview", |b| {
+ // Same as "zerovec_serde/deserialize_sum/u32/zerovec"
+ let buffer = bincode::serialize(
+ &ZeroVec::<u32>::parse_byte_slice(black_box(TEST_BUFFER_LE)).unwrap(),
+ )
+ .unwrap();
+ b.iter(|| {
+ bincode::deserialize::<ZeroVec<u32>>(&buffer)
+ .unwrap()
+ .iter()
+ .sum::<u32>()
+ });
+ });
+
+ #[cfg(feature = "bench")]
+ {
+ u32_benches(c);
+ char_benches(c);
+ stress_benches(c);
+ }
+}
+
+#[cfg(feature = "bench")]
+fn u32_benches(c: &mut Criterion) {
+ c.bench_function("zerovec_serde/serialize/u32/slice", |b| {
+ b.iter(|| bincode::serialize(&Vec::from(black_box(TEST_SLICE))));
+ });
+
+ c.bench_function("zerovec_serde/deserialize_sum/u32/slice", |b| {
+ let buffer = bincode::serialize(&Vec::from(black_box(TEST_SLICE))).unwrap();
+ b.iter(|| {
+ bincode::deserialize::<Vec<u32>>(&buffer)
+ .unwrap()
+ .iter()
+ .sum::<u32>()
+ });
+ });
+
+ c.bench_function("zerovec_serde/serialize/u32/zerovec", |b| {
+ b.iter(|| bincode::serialize(&ZeroVec::from_slice_or_alloc(black_box(TEST_SLICE))));
+ });
+
+ c.bench_function("zerovec_serde/deserialize_sum/u32/zerovec", |b| {
+ let buffer = bincode::serialize(
+ &ZeroVec::<u32>::parse_byte_slice(black_box(TEST_BUFFER_LE)).unwrap(),
+ )
+ .unwrap();
+ b.iter(|| {
+ bincode::deserialize::<ZeroVec<u32>>(&buffer)
+ .unwrap()
+ .iter()
+ .sum::<u32>()
+ });
+ });
+}
+
+#[cfg(feature = "bench")]
+fn char_benches(c: &mut Criterion) {
+ const ORIGINAL_CHARS: &[char] = &[
+ 'ⶢ', '⺇', 'Ⱜ', '◁', '◩', '⌂', '⼅', '⏻', '⢜', '◊', 'ⲫ', '⏷', '◢', '⟉', '℞',
+ ];
+
+ let char_zero_vec = &ZeroVec::alloc_from_slice(ORIGINAL_CHARS);
+
+ c.bench_function("zerovec_serde/serialize/char/slice", |b| {
+ b.iter(|| bincode::serialize(black_box(&Vec::from(ORIGINAL_CHARS))));
+ });
+
+ c.bench_function("zerovec_serde/deserialize/char/slice", |b| {
+ let buffer = bincode::serialize(black_box(&Vec::from(ORIGINAL_CHARS))).unwrap();
+ b.iter(|| bincode::deserialize::<Vec<char>>(&buffer));
+ });
+
+ c.bench_function("zerovec_serde/serialize/char/zerovec", |b| {
+ b.iter(|| bincode::serialize(black_box(char_zero_vec)));
+ });
+
+ c.bench_function("zerovec_serde/deserialize/char/zerovec", |b| {
+ let buffer = bincode::serialize(black_box(char_zero_vec)).unwrap();
+ b.iter(|| bincode::deserialize::<ZeroVec<char>>(&buffer));
+ });
+}
+
+#[cfg(feature = "bench")]
+fn stress_benches(c: &mut Criterion) {
+ let number_vec = random_numbers(100);
+ let bincode_vec = bincode::serialize(&number_vec).unwrap();
+ let zerovec_aligned = ZeroVec::from_slice_or_alloc(number_vec.as_slice());
+ let bincode_zerovec = bincode::serialize(&zerovec_aligned).unwrap();
+
+ // *** Deserialize vec of 100 `u32` ***
+ c.bench_function("zerovec_serde/deserialize/stress/vec", |b| {
+ b.iter(|| bincode::deserialize::<Vec<u32>>(&bincode_vec));
+ });
+
+ // *** Deserialize vec of 100 `u32` ***
+ c.bench_function("zerovec_serde/deserialize/stress/zerovec", |b| {
+ b.iter(|| bincode::deserialize::<ZeroVec<u32>>(&bincode_zerovec));
+ });
+
+ // *** Compute sum of vec of 100 `u32` ***
+ c.bench_function("zerovec_serde/sum/stress/vec", |b| {
+ b.iter(|| black_box(&number_vec).iter().sum::<u32>());
+ });
+
+ // *** Compute sum of vec of 100 `u32` ***
+ let zerovec = ZeroVec::<u32>::parse_byte_slice(zerovec_aligned.as_bytes()).unwrap();
+ c.bench_function("zerovec_serde/sum/stress/zerovec", |b| {
+ b.iter(|| black_box(&zerovec).iter().sum::<u32>());
+ });
+}
+
+criterion_group!(benches, overview_bench,);
+criterion_main!(benches);