tenferro-gpu 0.1.0

CubeCL-backed CUDA and WebGPU provider backends for tenferro tensors.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

use super::*;

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

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

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