Hugging Face's logo

Spaces:
reztilop
/
colibri.qdrant
Build error

App Files Community
colibri.qdrant
File size: 2,381 Bytes 
84d2a97
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
#[cfg(not(target_os = "windows"))]
mod prof;

use std::collections::{BTreeMap, HashMap};

use criterion::{criterion_group, criterion_main, Criterion};
use rand::rngs::StdRng;
use rand::SeedableRng;
use segment::data_types::tiny_map::TinyMap;
use segment::fixtures::index_fixtures::random_vector;

const DIM: usize = 100;

fn small_map_obj(c: &mut Criterion) {
    let mut group = c.benchmark_group("small-map-obj-group");

    let mut rng = StdRng::seed_from_u64(42);
    let default_key = "vector".to_string();
    let default_key_2 = "vector1".to_string();
    let default_key_3 = "vector2".to_string();
    let random_vector = random_vector(&mut rng, DIM);

    group.bench_function("hash-map", |b| {
        b.iter(|| {
            let mut map = HashMap::new();
            map.insert(default_key.clone(), random_vector.clone());
            map.insert(default_key_2.clone(), random_vector.clone());
            map.insert(default_key_3.clone(), random_vector.clone());
            let _ = map.get(&default_key_3);
        });
    });

    group.bench_function("btree-map", |b| {
        b.iter(|| {
            let mut map = BTreeMap::new();
            map.insert(default_key.clone(), random_vector.clone());
            map.insert(default_key_2.clone(), random_vector.clone());
            map.insert(default_key_3.clone(), random_vector.clone());
            let _ = map.get(&default_key_3);
        });
    });

    #[allow(clippy::vec_init_then_push)]
    group.bench_function("vec-map", |b| {
        b.iter(|| {
            let mut map = Vec::with_capacity(3);
            map.push((default_key.clone(), random_vector.clone()));
            map.push((default_key_2.clone(), random_vector.clone()));
            map.push((default_key_3.clone(), random_vector.clone()));
            let _ = map.iter().find(|(k, _)| k == &default_key_3);
        });
    });

    group.bench_function("tiny-map", |b| {
        b.iter(|| {
            let mut map = TinyMap::new();
            map.insert(default_key.clone(), random_vector.clone());
            map.insert(default_key_2.clone(), random_vector.clone());
            map.insert(default_key_3.clone(), random_vector.clone());
            let _ = map.get(&default_key_3);
        });
    });
}

criterion_group! {
    name = benches;
    config = Criterion::default();
    targets = small_map_obj
}

criterion_main!(benches);