kwaai-compression 0.4.114

Compression utilities for KwaaiNet - 8-bit quantization, gradient compression
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

//! Error types for compression operations

use thiserror::Error;

/// Result type for compression operations
pub type CompressionResult<T> = Result<T, CompressionError>;

/// Errors that can occur during compression
#[derive(Error, Debug)]
pub enum CompressionError {
    /// Compression failed
    #[error("Compression failed: {0}")]
    CompressionFailed(String),

    /// Decompression failed
    #[error("Decompression failed: {0}")]
    DecompressionFailed(String),

    /// Invalid data
    #[error("Invalid compressed data: {0}")]
    InvalidData(String),

    /// Shape mismatch
    #[error("Shape mismatch: expected {expected:?}, got {actual:?}")]
    ShapeMismatch {
        expected: Vec<usize>,
        actual: Vec<usize>,
    },

    /// Tensor error
    #[error("Tensor error: {0}")]
    TensorError(String),
}

impl From<candle_core::Error> for CompressionError {
    fn from(err: candle_core::Error) -> Self {
        CompressionError::TensorError(err.to_string())
    }
}