diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.rs | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/main.rs b/src/main.rs index 655c906..be0571d 100644 --- a/src/main.rs +++ b/src/main.rs @@ -4,10 +4,11 @@ use mdarray::{view, tensor, Slice, Dim, Const, expr::Expression}; use mdarray as md; // Indexing convention: C_ij <- A_ik * B_kj -fn matmul<D0: Dim, D1: Dim, D2: Dim>( - a: &Slice<f64, (D0, D1)>, - b: &Slice<f64, (D1, D2)>, - c: &mut Slice<f64, (D0, D2)>) { +fn matmul<DI: Dim, DK: Dim, DJ: Dim>( + a: &Slice<f64, (DI, DK)>, + b: &Slice<f64, (DK, DJ)>, + c: &mut Slice<f64, (DI, DJ)>, +) { for (mut ci, ai) in c.rows_mut().zip(a.rows()) { for (aik, bk) in ai.zip(b.rows()) { for (cij, bkj) in ci.expr_mut().zip(bk) { |