Skip to main content

diskann_disk/utils/
mod.rs

1/*
2 * Copyright (c) Microsoft Corporation.
3 * Licensed under the MIT license.
4 */
5
6//! Disk-specific utilities and helper functions.
7//!
8//! This module contains utility functions and helpers
9//! specific to disk index operations.
10
11pub mod partition;
12pub use partition::partition_with_ram_budget;
13
14pub mod math_util;
15pub use math_util::{compute_closest_centers, compute_closest_centers_in_block, compute_vecs_l2sq};
16
17pub mod kmeans;
18pub use kmeans::{k_means_clustering, k_meanspp_selecting_pivots, run_lloyds};
19
20pub mod instrumentation;
21
22pub mod aligned_file_reader;
23pub use aligned_file_reader::AlignedFileReaderFactory;
24#[cfg(any(feature = "virtual_storage", test))]
25pub use aligned_file_reader::VirtualAlignedReaderFactory;
26
27pub mod statistics;
28pub use statistics::{get_mean_stats, get_percentile_stats, get_sum_stats, QueryStatistics};