knok 0.1.0

Compile-time linalg graphs for Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

extern crate alloc;

pub type OutputF32 = alloc::vec::Vec<f32>;

pub fn invoke_f32(
    vmfb: &[u8],
    function_name: &'static str,
    backend: &'static str,
    inputs: &[(&[usize], &[f32])],
) -> crate::Result<OutputF32> {
    crate::runtime::invoke_f32(vmfb, function_name, backend, inputs)
}

pub fn invoke_f32_with_engine(
    engine: &crate::Engine,
    vmfb: &[u8],
    function_name: &'static str,
    backend: &'static str,
    driver: &'static str,
    inputs: &[(&[usize], &[f32])],
) -> crate::Result<OutputF32> {
    engine.invoke_raw_f32(vmfb, function_name, backend, driver, inputs)
}