[Mlir-commits] [mlir] [mlir][ArmSME] Add optional mask operand to tile_store (PR #70657)

Benjamin Maxwell llvmlistbot at llvm.org
Tue Oct 31 08:17:02 PDT 2023


================
@@ -356,6 +358,11 @@ def TileStoreOp : ArmSME_Op<"tile_store"> {
     rank 2 with dynamic 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 `mask` may be specified to mask out elements written
+    to the MemRef. The `mask` type is an `i1` vector of the same shape as the
+    vector type that matches how elements are written into the MemRef. Elements
+    whose corresponding mask element is `0` are masked out.
----------------
MacDue wrote:

nit: This seems a little too verbose to me:
```suggestion
    An optional `mask` may be provided, the shape of the mask corresponds to the `tile`,
    and selects which elements of the tile will be stored. 
```

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


More information about the Mlir-commits mailing list