Skip to main content

compile_flat

Function compile_flat 

Source
pub fn compile_flat<F>(f: F) -> impl Fn(&[Array])
where F: Fn(&[Array]) -> Result<Vec<Array>, Error>,
Expand description

compile over positional arrays: traces once per input signature (dtypes and shapes) and default stream.

use nuvec_compile::compile_flat;
use nuvec_core::{Array, Result, array};

let f = compile_flat(|xs: &[Array]| Ok(vec![xs[0].multiply(&xs[0]).add(1.0)]));
assert_eq!(f(&[array![1.0, 2.0]])?[0].to_vec::<f32>()?, [2.0, 5.0]);