Struct GpuVectorIndex 

pub struct GpuVectorIndex { /* private fields */ }

GPU-accelerated vector index

Implementations

impl GpuVectorIndex

pub fn new(config: GpuConfig) -> Result<Self>

Create a new GPU vector index

pub fn add_vectors(&mut self, vectors: Vec<Vector>) -> Result<()>

Add vectors to the index

pub fn search( &self, query: &Vector, k: usize, metric: SimilarityMetric, ) -> Result<Vec<(usize, f32)>>

Search for similar vectors

pub fn batch_search( &self, queries: &[Vector], k: usize, metric: SimilarityMetric, ) -> Result<Vec<Vec<(usize, f32)>>>

Batch search for multiple queries

pub fn len(&self) -> usize

Get the number of vectors in the index

pub fn is_empty(&self) -> bool

Check if the index is empty

pub fn dimension(&self) -> usize

Get the dimension of vectors in this index

pub fn performance_stats(&self) -> Arc<RwLock<GpuPerformanceStats>>

Get performance statistics

pub fn clear(&mut self)

Clear the index

Trait Implementations

impl Debug for GpuVectorIndex

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl VectorIndex for GpuVectorIndex

fn insert(&mut self, uri: String, vector: Vector) -> Result<()>

Insert a vector with associated URI

fn search_knn(&self, query: &Vector, k: usize) -> Result<Vec<(String, f32)>>

Find k nearest neighbors

fn search_threshold( &self, query: &Vector, threshold: f32, ) -> Result<Vec<(String, f32)>>

Find all vectors within threshold similarity

fn get_vector(&self, uri: &str) -> Option<&Vector>

Get a vector by its URI

fn add_vector( &mut self, id: VectorId, vector: Vector, _metadata: Option<HashMap<String, String>>, ) -> Result<()>

Add a vector with associated ID and metadata

fn update_vector(&mut self, id: VectorId, vector: Vector) -> Result<()>

Update an existing vector

fn update_metadata( &mut self, _id: VectorId, _metadata: HashMap<String, String>, ) -> Result<()>

Update metadata for a vector

fn remove_vector(&mut self, _id: VectorId) -> Result<()>

Remove a vector by its ID