Module gpu

Module gpu 

Source
Expand description

GPU acceleration for vector operations using CUDA

This module provides GPU acceleration for:

  • Distance calculations (cosine, euclidean, etc.)
  • Batch vector operations
  • Parallel search algorithms
  • Matrix operations for embeddings

Re-exports§

pub use accelerator::create_default_accelerator;
pub use accelerator::create_memory_optimized_accelerator;
pub use accelerator::create_performance_accelerator;
pub use accelerator::is_gpu_available;
pub use accelerator::GpuAccelerator;
pub use buffer::GpuBuffer;
pub use config::GpuConfig;
pub use config::OptimizationLevel;
pub use config::PrecisionMode;
pub use device::GpuDevice;
pub use index::AdvancedGpuVectorIndex;
pub use index::BatchVectorProcessor;
pub use index::GpuVectorIndex;
pub use memory_pool::GpuMemoryPool;
pub use performance::GpuPerformanceStats;
pub use types::GpuExecutionConfig;
pub use kernels::*;

Modules§

accelerator
Main GPU accelerator implementation
buffer
GPU memory buffer management
config
GPU configuration structures and enums
device
GPU device information and management
index
GPU-accelerated vector index implementations
kernels
CUDA kernel implementations for various vector operations
memory_pool
GPU memory pool management for efficient allocation and reuse
performance
GPU performance monitoring and statistics
runtime
CUDA runtime types and utilities
types
Common GPU types and structures