[Mlir-commits] [mlir] [mlir][ArmSME] Add optional padding and mask operands to tile_load (PR #69195)

Andrzej WarzyƄski llvmlistbot at llvm.org
Wed Oct 25 03:18:47 PDT 2023


================
@@ -242,6 +259,16 @@ def TileLoadOp : ArmSME_Op<"tile_load"> {
     dimensions, since the operation is scalable, and the element type must be a
     scalar that matches the element type of the result.
 
+    An optional SSA value `padding` of the same elemental type as the MemRef is
+    provided to specify a fallback value in the case of masking.
+
+    An optional SSA value `mask` may be specified to mask out elements read
+    from the MemRef. The `mask` type is an `i1` vector with a shape that
+    matches how elements are read from the MemRef. Elements whose corresponding
+    mask element is `0` are masked out and replaced with `padding`.
+
+    If either `padding` or `mask` are specified, both must be specified.
----------------
banach-space wrote:

Yeah, i think that's missing. If this is to require a custom verifier then I'd be happy to be implemented in a separate patch (to unblock tings).

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


More information about the Mlir-commits mailing list