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 types;
10pub mod traits;
11
12// Re-export commonly used types
13pub use error::{HiveGpuError, Result};
14pub use types::{
15    GpuVector, GpuDistanceMetric, GpuSearchResult, GpuDeviceInfo, 
16    GpuCapabilities, GpuMemoryStats, HnswConfig, VectorMetadata
17};
18pub use traits::{
19    GpuBackend, GpuVectorStorage, GpuContext, GpuBufferManager, 
20    GpuMonitor, GpuBuffer, BufferType, BufferPoolStats, VramStats, VramBufferInfo
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#[cfg(feature = "wgpu")]
31pub mod wgpu;
32
33// Backend detection
34pub mod backends;
35
36// Monitoring utilities
37pub mod monitoring;
38
39// Shader management
40pub mod shaders;
41
42// Utility functions
43pub mod utils;
44
45// Version information
46pub const VERSION: &str = env!("CARGO_PKG_VERSION");
47
48// Include test modules
49// #[cfg(test)]
50// mod tests;