[Mlir-commits] [mlir] [mlir][ArmSME] Add initial SME vector legalization pass (PR #79152)

Andrzej WarzyƄski llvmlistbot at llvm.org
Fri Jan 26 03:17:57 PST 2024


================
@@ -122,4 +122,23 @@ def TileAllocation
   let dependentDialects = ["func::FuncDialect"];
 }
 
+def VectorLegalization
+  : Pass<"arm-sme-vector-legalization", "mlir::ModuleOp"> {
+  let summary = "Legalize vectors for ArmSME";
+  let description = [{
+    This pass legalizes vector operations so that they can be lowered to ArmSME.
+    This includes decomposing operations that operate on vector types larger
+    than a single SME tile (e.g. `vector<[8]x[8]xf32>`) into multiple SME
----------------
banach-space wrote:

> legal - means something that can be lowered to ArmSME 

Perhaps you could try to defining that? I assume it means either:
* 2d scalable vectors with sizes matching SME virtual tiles,
* 1d scalable vectors
* n-D vectors with at most the trailing dim scalable?

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


More information about the Mlir-commits mailing list