[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