algeon-cuvs 0.1.3

Algeon-safe Rust bindings for NVIDIA cuVS GPU vector search and clustering
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
35
36
37
38

//! Runtime-bound safe Rust integration for the scoped cuVS surface used by Algeon.
//!
//! Published as an Algeon implementation dependency. Applications use
//! [`algeon-datafusion`](https://docs.rs/algeon-datafusion); workspace code
//! uses this crate for its component contracts. See the
//! [API documentation boundary](https://algeon.dev/design#api-documentation).

mod brute_force;
mod cagra;
mod device;
mod dlpack;
mod elements;
mod error;
mod ivf_flat;
mod ivf_pq;
mod kmeans;
mod pairwise;
mod params_handle;
mod pca;
mod prepared;
mod resources;
mod select_k;

pub use brute_force::{BruteForceIndex, DistanceMetric, KnnSearchOutput};
pub use cagra::{CagraBuildAlgo, CagraIndex, CagraIndexParams, CagraSearchAlgo, CagraSearchParams};
pub use device::{
    DeviceBuffer, DeviceElement, DeviceTensor, DeviceTensorMut, MatrixLayout, OwnedMatrix, ValidatedDeviceTensor,
};
pub use elements::{CagraElement, IvfElement, KMeansElement, PairwiseElement};
pub use error::{Error, ErrorKind, ErrorStatus, Result};
pub use ivf_flat::{IvfFlatIndex, IvfFlatIndexParams, IvfFlatSearchParams};
pub use ivf_pq::{IvfPqCodebookKind, IvfPqIndex, IvfPqIndexParams, IvfPqSearchParams};
pub use kmeans::{KMeansFit, KMeansInit, KMeansMetric, KMeansModel, KMeansParams, KMeansPrediction};
pub use pairwise::pairwise_distance_into;
pub use pca::{PcaFitTransform, PcaOutputBuffers, PcaParams, PcaSolver, fit_transform_owned};
pub use prepared::{PreparedTensorLease, PreparedValidatedTensorLease};
pub use resources::{CuvsResources, StreamPoolCoverage};
pub use select_k::select_k_into;