[Mlir-commits] [mlir] [mlir][vector] Extend mask calculation for vector.contract (PR #65733)

Andrzej WarzyƄski llvmlistbot at llvm.org
Mon Sep 11 02:18:01 PDT 2023


================
@@ -950,3 +979,22 @@ func.func @vector_scalable_extract(%sv: vector<[8]xi32>) {
   %2 = vector.scalable.extract %sv[4] : vector<4xi32> from vector<[8]xi32>
   return
  }
+
+#matmat_accesses = [
+  affine_map<(i, j, k) -> (i, k)>,
+  affine_map<(i, j, k) -> (k, j)>,
+  affine_map<(i, j, k) -> (i, j)>
+]
+#matmat_trait = {
+  indexing_maps = #matmat_accesses,
+  iterator_types = ["parallel", "parallel", "reduction"]
+}
+func.func @matmul_masked_scalable(%arg0: vector<3x4xf32>,
----------------
banach-space wrote:

Ta, see the latest fixup commit

https://github.com/llvm/llvm-project/pull/65733


More information about the Mlir-commits mailing list