Crate rustyhdf5_gpu 

GPU-accelerated vector operations for rustyhdf5.

Uses wgpu compute shaders (WGSL) for cross-platform GPU acceleration. Supports CUDA, Metal, Vulkan, and DirectX 12 backends via wgpu.

Usage

use rustyhdf5_gpu::GpuAccelerator;

// Fall back to CPU if GPU is not available
let mut gpu = match GpuAccelerator::new() {
    Ok(g) => g,
    Err(_) => { /* use CPU path */ return; }
};

// Upload vectors once
let vectors = vec![0.0f32; 1000 * 128]; // 1000 vectors of dim 128
let norms = vec![1.0f32; 1000];
gpu.upload_vectors(&vectors, 128).unwrap();
gpu.upload_norms(&norms).unwrap();

// Search many times
let query = vec![1.0f32; 128];
let results = gpu.cosine_search(&query, 10).unwrap();

Re-exports

pub use device::DeviceInfo;

pub use error::GpuError;

pub use error::Result;

Modules

device

error

Structs

GpuAccelerator

GPU-accelerated vector search engine.