gemm/
gemm.rs

1use custos_math::{custos::CPU, Matrix};
2
3fn main() {
4    let device = CPU::new();
5
6    let a = Matrix::from((&device, (2, 3), [1., 2., 3., 4., 5., 6.]));
7    let b = Matrix::from((&device, (3, 2), [6., 5., 4., 3., 2., 1.]));
8
9    let c = a.gemm(&b);
10    println!("c: {c:?}");
11
12    assert_eq!(c.read(), vec![20., 14., 56., 41.,]);
13}