diskann-benchmark-runner 0.56.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

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

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

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,
}