numpy 0.29.0

PyO3-based Rust bindings of the NumPy C-API
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

use std::hint::black_box;

use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion, Throughput};
use numpy::{PyArrayLike1, PyArrayLike2, PyArrayLike3};
use pyo3::{
    ffi::c_str,
    types::{PyAnyMethods, PyDict},
    Python,
};

fn extract_array_like_1(c: &mut Criterion) {
    const SIZES: &[usize] = &[2_usize.pow(5), 2_usize.pow(10), 2_usize.pow(15)];

    let mut group = c.benchmark_group("extract_array_like_1");
    for &size in SIZES {
        Python::attach(|py| {
            let locals = PyDict::new(py);
            locals.set_item("size", size).unwrap();

            let list = py
                .eval(
                    c_str!("[float(i) for i in range(size)]"),
                    Some(&locals),
                    None,
                )
                .unwrap();

            group.throughput(Throughput::Elements(size as u64));
            group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, _size| {
                b.iter(|| {
                    let list = black_box(&list);

                    let _array: PyArrayLike1<'_, f64> = black_box(list.extract().unwrap());
                });
            });
        });
    }
}

fn extract_array_like_2(c: &mut Criterion) {
    const SIZES: &[usize] = &[2_usize.pow(3), 2_usize.pow(5), 2_usize.pow(8)];

    let mut group = c.benchmark_group("extract_array_like_2");
    for &size in SIZES {
        Python::attach(|py| {
            let locals = PyDict::new(py);
            locals.set_item("size", size).unwrap();

            let list = py
                .eval(
                    c_str!("[[float(i + j) for i in range(size)] for j in range(size)]"),
                    Some(&locals),
                    None,
                )
                .unwrap();

            group.throughput(Throughput::Elements(size.pow(2) as u64));
            group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, _size| {
                b.iter(|| {
                    let list = black_box(&list);

                    let _array: PyArrayLike2<'_, f64> = black_box(list.extract().unwrap());
                });
            });
        });
    }
}

fn extract_array_like_3(c: &mut Criterion) {
    const SIZES: &[usize] = &[2_usize.pow(2), 2_usize.pow(4), 2_usize.pow(5)];

    let mut group = c.benchmark_group("extract_array_like_3");
    for &size in SIZES {
        Python::attach(|py| {
            let locals = PyDict::new(py);
            locals.set_item("size", size).unwrap();

            let list = py
                .eval(
                    c_str!("[[[float(i + j + k) for i in range(size)] for j in range(size)] for k in range(size)]"),
                    Some(&locals),
                    None,
                )
                .unwrap();

            group.throughput(Throughput::Elements(size.pow(3) as u64));
            group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, _size| {
                b.iter(|| {
                    let list = black_box(&list);

                    let _array: PyArrayLike3<'_, f64> = black_box(list.extract().unwrap());
                });
            });
        });
    }
}

criterion_group!(
    benches,
    extract_array_like_1,
    extract_array_like_2,
    extract_array_like_3
);
criterion_main!(benches);