hive_gpu/
lib.rs

1//! Hive GPU - High-performance GPU acceleration for vector operations
2//!
3//! This crate provides GPU-accelerated vector operations using Metal (Apple Silicon)
4//! and CUDA (NVIDIA) backends for maximum performance in vector similarity search.
5
6#![allow(warnings)]
7
8pub mod error;
9pub mod traits;
10pub mod types;
11
12// Re-export commonly used types
13pub use error::{HiveGpuError, Result};
14pub use traits::{
15    BufferPoolStats, BufferType, GpuBackend, GpuBuffer, GpuBufferManager, GpuContext, GpuMonitor,
16    GpuVectorStorage, VramBufferInfo, VramStats,
17};
18pub use types::{
19    GpuCapabilities, GpuDeviceInfo, GpuDistanceMetric, GpuMemoryStats, GpuSearchResult, GpuVector,
20    HnswConfig, VectorMetadata,
21};
22
23// Platform-specific modules
24#[cfg(all(target_os = "macos", feature = "metal-native"))]
25pub mod metal;
26
27#[cfg(feature = "cuda")]
28pub mod cuda;
29
30// Backend detection
31pub mod backends;
32
33// Monitoring utilities
34pub mod monitoring;
35
36// Shader management
37pub mod shaders;
38
39// Utility functions
40pub mod utils;
41
42// Version information
43pub const VERSION: &str = env!("CARGO_PKG_VERSION");
44
45// Include test modules
46// #[cfg(test)]
47// mod tests;