diskann-benchmark-runner 0.60.0

DiskANN3 is a composable library for bringing scalable, accurate and cost-effective vector indexing to multiple databases.
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.
 */

//! Development CLI for exercising the benchmark runner with its test registry.

use clap::Parser;
use diskann_benchmark_runner::{App, Registry, output, test};

fn main() -> anyhow::Result<()> {
    // Parse the command line options.
    let cli = Cli::parse();

    let mut registry = Registry::new();
    let config = test::TestConfig::with_features(cli.features);
    test::register_benchmarks(&mut registry, &config)?;

    cli.app.run(&registry, &mut output::default())
}

/// Top level CLI.
///
/// In addition to the `App`, we accept a list of features for the `TestConfig`.
#[derive(Debug, clap::Parser)]
struct Cli {
    /// Emulate enabling various features for feature gated functionality.
    #[arg(long, value_delimiter = ',', num_args = 0..)]
    features: Vec<String>,

    /// The actual application.
    #[command(flatten)]
    app: App,
}