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
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

use std::error::Error as _;

use super::*;

#[test]
fn unsupported_dtype_preserves_classification_and_source() {
    let error = unsupported_dtype("exp", DType::I32);

    assert_eq!(error.kind(), ErrorKind::Unsupported);
    let source = error.source().expect("extension errors have a source");
    let source = source
        .downcast_ref::<CudaError>()
        .expect("CUDA errors preserve their typed source");
    assert!(matches!(
        source,
        CudaError::UnsupportedDType {
            op: "exp",
            dtype: DType::I32
        }
    ));
}

#[test]
fn provider_status_preserves_classification_and_source() {
    let error = provider_status("dot_general", "cuTENSOR", "cutensorContract", 7);

    assert_eq!(error.kind(), ErrorKind::BackendFailure);
    let source = error.source().expect("CUDA errors have a source");
    let source = source
        .downcast_ref::<CudaError>()
        .expect("CUDA errors preserve their typed source");
    assert!(matches!(
        source,
        CudaError::ProviderStatus {
            library: "cuTENSOR",
            call: "cutensorContract",
            status: 7,
        }
    ));
}

#[test]
fn workspace_overflow_preserves_classification_and_source() {
    let error = workspace_size_overflow("dot_general", u64::MAX);

    assert_eq!(error.kind(), ErrorKind::BackendFailure);
    let source = error.source().expect("CUDA errors have a source");
    let source = source
        .downcast_ref::<CudaError>()
        .expect("CUDA errors preserve their typed source");
    assert!(matches!(
        source,
        CudaError::WorkspaceSizeOverflow {
            op: "dot_general",
            size: u64::MAX,
        }
    ));
}