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