diskann-disk 0.60.0

DiskANN3 is a composable library for bringing scalable, accurate and cost-effective vector indexing to multiple databases.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

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

use diskann::{utils::VectorRepr, ANNResult};
use diskann_utils::views::rowmajor;

/// A quantizer constructed once and shared across compression batches.
pub trait QuantCompressor<T>: Sized + Sync
where
    T: VectorRepr,
{
    type CompressorContext;

    /// Construct a quantizer, including any training and persistence it requires.
    ///
    /// # Errors
    /// Returns an error if construction fails.
    fn new(context: &Self::CompressorContext) -> ANNResult<Self>;

    fn compress(&self, vector: rowmajor::Ref<f32>, output: rowmajor::Mut<u8>) -> ANNResult<()>;
    fn compressed_bytes(&self) -> usize;
}