Skip to content

Commit f3ee1de

Browse files
committed
adjust benchmark
Signed-off-by: Alexander Droste <[email protected]>
1 parent 2d936cc commit f3ee1de

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

encodings/alp/benches/alp_compress.rs

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@
66
use divan::Bencher;
77
use rand::rngs::StdRng;
88
use rand::{Rng, SeedableRng as _};
9-
use vortex_alp::{ALPFloat, ALPRDFloat, RDEncoder, alp_encode};
9+
use vortex_alp::{ALPFloat, ALPRDFloat, RDEncoder, alp_encode, decompress};
1010
use vortex_array::arrays::PrimitiveArray;
1111
use vortex_array::compute::warm_up_vtables;
1212
use vortex_array::validity::Validity;
13-
use vortex_buffer::buffer;
13+
use vortex_buffer::{Buffer, buffer};
1414
use vortex_dtype::NativePType;
1515

1616
fn main() {
@@ -84,10 +84,15 @@ fn decompress_alp<T: ALPFloat + NativePType>(bencher: Bencher, args: (usize, f64
8484
Validity::NonNullable
8585
};
8686
let values = values.freeze();
87-
let array = alp_encode(&PrimitiveArray::new(values, validity), None).unwrap();
8887
bencher
89-
.with_inputs(|| array.clone())
90-
.bench_values(|array| array.to_canonical());
88+
.with_inputs(|| {
89+
alp_encode(
90+
&PrimitiveArray::new(Buffer::copy_from(&values), validity.clone()),
91+
None,
92+
)
93+
.unwrap()
94+
})
95+
.bench_values(|array| decompress(&array));
9196
}
9297

9398
#[divan::bench(types = [f32, f64], args = [10_000, 100_000])]

0 commit comments

Comments
 (0)