oxirs_vec/gpu/
mod.rs

1//! GPU acceleration for vector operations using CUDA
2//!
3//! This module provides GPU acceleration for:
4//! - Distance calculations (cosine, euclidean, etc.)
5//! - Batch vector operations
6//! - Parallel search algorithms
7//! - Matrix operations for embeddings
8
9pub mod accelerator;
10pub mod buffer;
11pub mod config;
12pub mod device;
13pub mod index;
14pub mod kernels;
15pub mod memory_pool;
16pub mod performance;
17pub mod runtime;
18pub mod types;
19
20// Re-export key types for convenience
21pub use accelerator::{
22    create_default_accelerator, create_memory_optimized_accelerator,
23    create_performance_accelerator, is_gpu_available, GpuAccelerator,
24};
25pub use buffer::GpuBuffer;
26pub use config::{GpuConfig, OptimizationLevel, PrecisionMode};
27pub use device::GpuDevice;
28pub use index::{AdvancedGpuVectorIndex, BatchVectorProcessor, GpuVectorIndex};
29pub use kernels::*;
30pub use memory_pool::GpuMemoryPool;
31pub use performance::GpuPerformanceStats;
32pub use types::GpuExecutionConfig;