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

use crate::device::DeviceElement;

mod sealed {
    pub trait Cagra {}
    pub trait Ivf {}
    pub trait Pairwise {}
    pub trait KMeans {}

    impl Cagra for f32 {}
    impl Cagra for i8 {}
    impl Cagra for u8 {}
    impl Ivf for f32 {}
    impl Ivf for i8 {}
    impl Ivf for u8 {}
    impl Pairwise for f32 {}
    impl Pairwise for f64 {}
    impl KMeans for f32 {}
    impl KMeans for f64 {}
}

pub trait CagraElement: DeviceElement + sealed::Cagra {}
impl<T: DeviceElement + sealed::Cagra> CagraElement for T {}

pub trait IvfElement: DeviceElement + sealed::Ivf {}
impl<T: DeviceElement + sealed::Ivf> IvfElement for T {}

pub trait PairwiseElement: DeviceElement + sealed::Pairwise {
    type Distance: DeviceElement;
}
impl PairwiseElement for f32 {
    type Distance = f32;
}
impl PairwiseElement for f64 {
    type Distance = f64;
}

pub trait KMeansElement: DeviceElement + sealed::KMeans {}
impl<T: DeviceElement + sealed::KMeans> KMeansElement for T {}