Add linalg mlir version of matmul 8x8 kernel

opencompl · Nov 1, 2023 · 7f41e83 · 7f41e83
1 parent 762ad8f
commit 7f41e83
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/kernels/matmul/8x8xf64/Makefile b/kernels/matmul/8x8xf64/Makefile
@@ -4,5 +4,6 @@ include ../../../snitch/Makefile.rules
 
 TESTS =
 TESTS += baseline.x
+TESTS += linalg.x
 
 include ../../Makefile.kernels
diff --git a/kernels/matmul/8x8xf64/linalg.mlir b/kernels/matmul/8x8xf64/linalg.mlir
@@ -0,0 +1,12 @@
+func.func public @matmul(%X: memref<8x8xf64>,
+                         %Y: memref<8x8xf64>,
+                         %Z: memref<8x8xf64>) {
+  "linalg.generic"(%X, %Y, %Z) ({
+  ^bb0(%x: f64, %y: f64, %z: f64):
+    %r0 = arith.mulf %x, %y : f64
+    %r1 = arith.addf %z, %r0 : f64
+    "linalg.yield"(%r1) : (f64) -> ()
+  }) {indexing_maps = [affine_map<(m, n, k) -> (m, k)>, affine_map<(m, n, k) -> (k, n)>, affine_map<(m, n, k) -> (m,
+  n)>], iterator_types = [#linalg.iterator_type<parallel>, #linalg.iterator_type<parallel>, #linalg.iterator_type<reduction>], operand_segment_sizes = array<i32: 2, 1>} : (memref<8x8xf64>, memref<8x8xf64>, memref<8x8xf64>) -> ()
+  func.return
+}
diff --git a/kernels/matmul/8x8xf64/linalg.xdsl.mlir b/kernels/matmul/8x8xf64/linalg.xdsl.mlir
@@ -0,0 +1,11 @@
+func.func public @matmul(%X: memref<8x8xf64>,
+                         %Y: memref<8x8xf64>,
+                         %Z: memref<8x8xf64>) {
+  "linalg.generic"(%X, %Y, %Z) ({
+  ^bb0(%x: f64, %y: f64, %z: f64):
+    %r0 = arith.mulf %x, %y : f64
+    %r1 = arith.addf %z, %r0 : f64
+    "linalg.yield"(%r1) : (f64) -> ()
+  }) {indexing_maps = [affine_map<(m, n, k) -> (m, k)>, affine_map<(m, n, k) -> (k, n)>, affine_map<(m, n, k) -> (m, n)>], iterator_types = [#linalg.iterator_type<parallel>, #linalg.iterator_type<parallel>, #linalg.iterator_type<reduction>], operandSegmentSizes = array<i32: 2, 1>} : (memref<8x8xf64>, memref<8x8xf64>, memref<8x8xf64>) -> ()
+  func.return
+}