libmir-metal 0.1.0

Metal inference backend for libmir
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use super::{Array, Result, Stream};

impl Array {
    pub fn add_scalar(&self, scalar: f32, stream: &Stream) -> Result<Self> {
        Self::from_native(stream.native().graph().add_scalar(self.native(), scalar)?)
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn adjusts_tensor_values_on_the_gpu_stream() -> Result<()> {
        let stream = Stream::new_gpu()?;
        let input = Array::from_f32(&[1.0, 2.0], &[1, 2])?;
        let output = input.add_scalar(1.0, &stream)?;
        output.async_eval()?;
        stream.synchronize()?;
        assert_eq!(output.to_vec_f32()?, vec![2.0, 3.0]);
        Ok(())
    }
}