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

use super::compact_col_major_strides;
use tenferro_tensor::{ErrorKind, ValidationKind};

const OP: &str = "native_permutation_test";

#[test]
fn compact_col_major_strides_validate_shape_metadata_once() {
    assert_eq!(
        compact_col_major_strides(OP, &[]).unwrap(),
        Vec::<isize>::new()
    );
    assert_eq!(compact_col_major_strides(OP, &[2, 3]).unwrap(), vec![1, 2]);

    let dimension = usize::MAX;
    let error = compact_col_major_strides("compact_stride_test", &[dimension]).unwrap_err();
    assert_eq!(
        error.kind(),
        ErrorKind::Validation(ValidationKind::InvalidArgument)
    );
    assert!(matches!(
        error,
        crate::Error::Validation {
            op: "compact_stride_test",
            source: tenferro_tensor::ValidationError::InvalidArgument {
                argument: "shape",
                message,
            },
        } if message == format!("dimension {dimension} cannot be represented as isize")
    ));

    let shape = [isize::MAX as usize, 2];
    let error = compact_col_major_strides("compact_stride_test", &shape).unwrap_err();
    assert_eq!(
        error.kind(),
        ErrorKind::Validation(ValidationKind::InvalidArgument)
    );
    assert!(matches!(
        error,
        crate::Error::Validation {
            op: "compact_stride_test",
            source: tenferro_tensor::ValidationError::InvalidArgument {
                argument: "shape",
                message,
            },
        } if message == format!("column-major stride overflow for shape {shape:?}")
    ));
}