diskann-benchmark-runner 0.51.0

DiskANN is a fast approximate nearest neighbor search library for high dimensional data
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

/*
 * Copyright (c) Microsoft Corporation.
 * Licensed under the MIT license.
 */

use crate::registry;

// submodules
mod dim;
mod typed;

pub(crate) use typed::TypeInput;

/////////
// API //
/////////

pub fn register_inputs(inputs: &mut registry::Inputs) -> anyhow::Result<()> {
    inputs.register::<typed::TypeInput>()?;
    inputs.register::<dim::DimInput>()?;
    Ok(())
}

pub fn register_benchmarks(benchmarks: &mut registry::Benchmarks) {
    benchmarks.register_regression("type-bench-f32", typed::TypeBench::<f32>::new());
    benchmarks.register_regression("type-bench-i8", typed::TypeBench::<i8>::new());
    benchmarks.register_regression(
        "exact-type-bench-f32-1000",
        typed::ExactTypeBench::<f32, 1000>::new(),
    );

    benchmarks.register("simple-bench", dim::SimpleBench);
    benchmarks.register_regression("dim-bench", dim::DimBench);
}