diskann-benchmark 0.51.0

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

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

use diskann_benchmark_runner::registry::Benchmarks;

cfg_if::cfg_if! {
    if #[cfg(feature = "disk-index")] {
        mod benchmarks;
        mod build;
        mod search;
        mod json_spancollector;

        /// Register disk index benchmarks when the `disk-index` feature is enabled.
        pub(crate) fn register_benchmarks(registry: &mut Benchmarks) {
            benchmarks::register_benchmarks(registry);
        }
    } else {
        crate::utils::stub_impl!(
            "disk-index",
            inputs::disk::DiskIndexOperation
        );

        /// Register a stub that guides users to enable the `disk-index` feature.
        pub(crate) fn register_benchmarks(registry: &mut Benchmarks) {
            imp::register("disk-index", registry);
        }
    }
}