[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