tenferro-gpu 0.3.0

CubeCL-backed CUDA and WebGPU provider backends for tenferro tensors.
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

use thiserror::Error;

/// Error returned by tenferro CubeCL kernel launch helpers.
#[derive(Clone, Debug, Error, PartialEq, Eq)]
pub(crate) enum CubeclKernelError {
    /// The requested axis is outside the input tensor rank.
    #[error("axis {axis} is out of bounds for rank {rank}")]
    InvalidAxis {
        /// Requested reduction axis.
        axis: usize,
        /// Input tensor rank.
        rank: usize,
    },

    /// The caller supplied an output tensor with a shape that is not keepdims.
    #[error("output shape {actual:?} does not match expected keepdims shape {expected:?}")]
    MismatchOutputShape {
        /// Expected output shape after setting the reduced axis length to one.
        expected: Vec<usize>,
        /// Actual output shape supplied by the caller.
        actual: Vec<usize>,
    },
    /// The selected launch strategy is invalid for the input.
    #[error("invalid reduction strategy: {reason}")]
    InvalidStrategy {
        /// Human-readable reason the strategy cannot be used.
        reason: String,
    },
}

/// Result alias for tenferro CubeCL kernel helpers.
pub(crate) type Result<T> = core::result::Result<T, CubeclKernelError>;