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 super::*;

#[test]
fn gpu_descriptors_have_catalog_dtype_policy() {
    let add = gpu_descriptor(PrimitiveOpKind::Add).unwrap();
    assert_eq!(add.dtype_policy, DTypePolicy::SameNumeric);
    assert_eq!(add.launch, GpuLaunchKind::BinaryFloatComplexInt);

    let div = gpu_descriptor(PrimitiveOpKind::Div).unwrap();
    assert_eq!(div.launch, GpuLaunchKind::BinaryFloatComplexInt);

    let rem = gpu_descriptor(PrimitiveOpKind::Rem).unwrap();
    assert_eq!(rem.launch, GpuLaunchKind::BinaryFloatInt);

    let pow = gpu_descriptor(PrimitiveOpKind::Pow).unwrap();
    assert_eq!(pow.launch, GpuLaunchKind::BinaryFloatInt);

    let compare = gpu_descriptor(PrimitiveOpKind::Compare).unwrap();
    assert_eq!(compare.dtype_policy, DTypePolicy::CompareToBool);
    assert_eq!(compare.launch, GpuLaunchKind::CompareFloatIntToBool);

    let select = gpu_descriptor(PrimitiveOpKind::Select).unwrap();
    assert_eq!(select.launch, GpuLaunchKind::SelectBoolFloatInt);
}

#[test]
fn host_only_primitives_have_no_gpu_descriptor() {
    assert_eq!(gpu_descriptor(PrimitiveOpKind::ShapeOf), None);
    assert_eq!(gpu_descriptor(PrimitiveOpKind::Constant), None);
}