ruda-kernel 0.1.2

Ruda Rust kernel DSL and device tensor operations.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

use crate::dsl::Runtime;
use ruda_core::tensor::DType;
use ruda_core::ir::features::TypeUsage;

pub fn supports_dtype<R: Runtime>(device: &R::Device, dtype: DType) -> bool {
    let client = R::client(device);

    let type_usage = client.properties().type_usage(dtype.into());
    // Same as `TypeUsage::all_scalar()`, but we make the usage explicit here
    type_usage.is_superset(
        TypeUsage::Buffer
            | TypeUsage::Conversion
            | TypeUsage::Arithmetic
            | TypeUsage::DotProduct,
    )
}