hive-gpu 0.2.0

High-performance GPU acceleration for vector operations with Device Info API (Metal, CUDA, ROCm)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

//! Error types for Hive GPU

use thiserror::Error;

/// Main error type for Hive GPU operations
#[derive(Error, Debug)]
pub enum HiveGpuError {
    /// Invalid vector dimension
    #[error("Invalid dimension: expected {expected}, got {got}")]
    InvalidDimension { expected: usize, got: usize },

    /// Dimension mismatch
    #[error("Dimension mismatch: expected {expected}, got {actual}")]
    DimensionMismatch { expected: usize, actual: usize },

    /// VRAM limit exceeded
    #[error("VRAM limit exceeded: requested {requested}, limit {limit}")]
    VramLimitExceeded { requested: usize, limit: usize },

    /// GPU operation failed
    #[error("GPU operation failed: {0}")]
    GpuOperationFailed(String),

    /// No GPU device available
    #[error("No GPU device available")]
    NoDeviceAvailable,

    /// Buffer allocation failed
    #[error("Buffer allocation failed: {0}")]
    BufferAllocationFailed(String),

    /// Device initialization failed
    #[error("Device initialization failed: {0}")]
    DeviceInitializationFailed(String),

    /// Shader compilation failed
    #[error("Shader compilation failed: {0}")]
    ShaderCompilationFailed(String),

    /// Memory allocation failed
    #[error("Memory allocation failed: {0}")]
    MemoryAllocationFailed(String),

    /// Search operation failed
    #[error("Search operation failed: {0}")]
    SearchFailed(String),

    /// Vector not found
    #[error("Vector not found: {0}")]
    VectorNotFound(String),

    /// Invalid configuration
    #[error("Invalid configuration: {0}")]
    InvalidConfiguration(String),

    /// CUDA driver or runtime failure
    #[error("CUDA error: {0}")]
    CudaError(String),

    /// cuBLAS call failure
    #[error("cuBLAS error: {0}")]
    CublasError(String),

    /// HIP driver or runtime failure
    #[error("HIP error: {0}")]
    HipError(String),

    /// rocBLAS call failure
    #[error("rocBLAS error: {0}")]
    RocblasError(String),

    /// Generic ROCm backend error (loader, setup, teardown)
    #[error("ROCm error: {0}")]
    RocmError(String),

    /// Vulkan API failure — covers Intel, universal-Vulkan fallback, and
    /// any host running a Vulkan-capable GPU through the `intel` feature.
    #[error("Vulkan error: {0}")]
    VulkanError(String),

    /// Generic Intel backend error (loader, device selection).
    #[error("Intel backend error: {0}")]
    IntelError(String),

    /// SPIR-V shader module creation or compilation failure.
    #[error("SPIR-V compile error: {0}")]
    SpirvCompileError(String),

    /// Internal error
    #[error("Internal error: {0}")]
    InternalError(String),

    /// IO error
    #[error("IO error: {0}")]
    IoError(#[from] std::io::Error),

    /// JSON error
    #[error("JSON error: {0}")]
    JsonError(#[from] serde_json::Error),

    /// Other errors
    #[error("{0}")]
    Other(String),
}

/// Result type alias for Hive GPU operations
pub type Result<T> = std::result::Result<T, HiveGpuError>;