[go: up one dir, main page]

arrow2 0.6.1

Unofficial implementation of Apache Arrow spec in safe Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

extern crate arrow2;

use arrow2::{
    compute::concat,
    datatypes::DataType,
    util::bench_util::{create_boolean_array, create_primitive_array},
};

use criterion::{criterion_group, criterion_main, Criterion};

fn add_benchmark(c: &mut Criterion) {
    (20..=20).step_by(2).for_each(|log2_size| {
        let size = 2usize.pow(log2_size);

        let array1 = create_primitive_array::<i32>(8, DataType::Int32, 0.5);
        let array2 = create_primitive_array::<i32>(size + 1, DataType::Int32, 0.5);

        c.bench_function(&format!("int32 concat aligned 2^{}", log2_size), |b| {
            b.iter(|| {
                let _ = concat::concatenate(&[&array1, &array2]);
            })
        });

        let array1 = create_primitive_array::<i32>(9, DataType::Int32, 0.5);

        c.bench_function(&format!("int32 concat unaligned 2^{}", log2_size), |b| {
            b.iter(|| {
                let _ = concat::concatenate(&[&array1, &array2]);
            })
        });

        let array1 = create_boolean_array(8, 0.5, 0.5);
        let array2 = create_boolean_array(size + 1, 0.5, 0.5);

        c.bench_function(&format!("boolean concat aligned 2^{}", log2_size), |b| {
            b.iter(|| {
                let _ = concat::concatenate(&[&array1, &array2]);
            })
        });

        let array1 = create_boolean_array(9, 0.5, 0.5);

        c.bench_function(&format!("boolean concat unaligned 2^{}", log2_size), |b| {
            b.iter(|| {
                let _ = concat::concatenate(&[&array1, &array2]);
            })
        });
    });
}

criterion_group!(benches, add_benchmark);
criterion_main!(benches);