rustyhdf5_gpu/lib.rs
1//! GPU-accelerated vector operations for rustyhdf5.
2//!
3//! Uses wgpu compute shaders (WGSL) for cross-platform GPU acceleration.
4//! Supports CUDA, Metal, Vulkan, and DirectX 12 backends via wgpu.
5//!
6//! # Usage
7//!
8//! ```no_run
9//! use rustyhdf5_gpu::GpuAccelerator;
10//!
11//! // Fall back to CPU if GPU is not available
12//! let mut gpu = match GpuAccelerator::new() {
13//! Ok(g) => g,
14//! Err(_) => { /* use CPU path */ return; }
15//! };
16//!
17//! // Upload vectors once
18//! let vectors = vec![0.0f32; 1000 * 128]; // 1000 vectors of dim 128
19//! let norms = vec![1.0f32; 1000];
20//! gpu.upload_vectors(&vectors, 128).unwrap();
21//! gpu.upload_norms(&norms).unwrap();
22//!
23//! // Search many times
24//! let query = vec![1.0f32; 128];
25//! let results = gpu.cosine_search(&query, 10).unwrap();
26//! ```
27
28pub mod device;
29pub mod error;
30
31#[cfg(feature = "gpu-wgpu")]
32mod accelerator;
33#[cfg(feature = "gpu-wgpu")]
34mod helpers;
35#[cfg(feature = "gpu-wgpu")]
36mod shaders;
37
38pub use device::DeviceInfo;
39pub use error::{GpuError, Result};
40
41#[cfg(feature = "gpu-wgpu")]
42pub use accelerator::GpuAccelerator;
43
44/// Stub when compiled without GPU support.
45#[cfg(not(feature = "gpu-wgpu"))]
46pub struct GpuAccelerator;
47
48#[cfg(not(feature = "gpu-wgpu"))]
49impl GpuAccelerator {
50 pub fn is_available() -> bool {
51 false
52 }
53
54 pub fn new() -> Result<Self> {
55 Err(GpuError::NotCompiled)
56 }
57}