pub struct GpuAccelerator { /* private fields */ }Expand description
GPU acceleration engine for vector operations
Implementations§
Source§impl GpuAccelerator
impl GpuAccelerator
pub fn new(config: GpuConfig) -> Result<Self>
Sourcepub fn compute_similarity(
&self,
queries: &[f32],
database: &[f32],
query_count: usize,
db_count: usize,
dim: usize,
metric: SimilarityMetric,
) -> Result<Vec<f32>>
pub fn compute_similarity( &self, queries: &[f32], database: &[f32], query_count: usize, db_count: usize, dim: usize, metric: SimilarityMetric, ) -> Result<Vec<f32>>
Compute similarity between query vectors and database vectors
Sourcepub fn performance_stats(&self) -> Arc<RwLock<GpuPerformanceStats>> ⓘ
pub fn performance_stats(&self) -> Arc<RwLock<GpuPerformanceStats>> ⓘ
Get performance statistics
Sourcepub fn synchronize(&self) -> Result<()>
pub fn synchronize(&self) -> Result<()>
Synchronize all operations
Sourcepub fn reset_stats(&self)
pub fn reset_stats(&self)
Reset performance statistics
Sourcepub fn get_memory_usage(&self) -> Result<usize>
pub fn get_memory_usage(&self) -> Result<usize>
Get current GPU memory usage in bytes
Trait Implementations§
Source§impl Debug for GpuAccelerator
impl Debug for GpuAccelerator
Source§impl Drop for GpuAccelerator
impl Drop for GpuAccelerator
impl Send for GpuAccelerator
impl Sync for GpuAccelerator
Auto Trait Implementations§
impl Freeze for GpuAccelerator
impl !RefUnwindSafe for GpuAccelerator
impl Unpin for GpuAccelerator
impl !UnwindSafe for GpuAccelerator
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<T> PolicyExt for Twhere
T: ?Sized,
impl<T> PolicyExt for Twhere
T: ?Sized,
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.