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

use crate::{DType, Error, ErrorKind};

#[derive(Debug, thiserror::Error)]
pub(crate) enum WebGpuError {
    #[error("{op} does not support dtype {dtype:?} on WebGPU")]
    UnsupportedDType { op: &'static str, dtype: DType },
    #[error("{op} is unsupported on WebGPU: {detail}")]
    UnsupportedOperation {
        op: &'static str,
        detail: &'static str,
    },
}

pub(crate) fn unsupported_dtype(op: &'static str, dtype: DType) -> Error {
    Error::extension(
        op,
        "webgpu",
        ErrorKind::Unsupported,
        WebGpuError::UnsupportedDType { op, dtype },
    )
}

pub(crate) fn unsupported_operation(op: &'static str, detail: &'static str) -> Error {
    Error::extension(
        op,
        "webgpu",
        ErrorKind::Unsupported,
        WebGpuError::UnsupportedOperation { op, detail },
    )
}