cpu_device/
cpu_device.rs

1use custos_math::{
2    custos::{Read, CPU},
3    BaseOps, Matrix,
4};
5
6fn main() {
7    let device = CPU::new();
8    let a = Matrix::<f32>::new(&device, (5, 5));
9    let b = Matrix::from((&device, (5, 5), vec![1.3; 5 * 5]));
10
11    let out = device.add(&a, &b);
12
13    assert_eq!(device.read(&out), vec![1.3; 5 * 5]);
14}