[Mlir-commits] [mlir] [mlir][ArmSME] Add mask operand to store_tile_slice (PR #70838)

Benjamin Maxwell llvmlistbot at llvm.org
Wed Nov 1 05:18:37 PDT 2023


================
@@ -489,22 +492,27 @@ def StoreTileSliceOp : ArmSME_Op<"store_tile_slice"> {
     dimensions since the operation is scalable, and the element type must be a
     scalar that matches the element type of the input tile.
 
+    An SSA value `mask` specifies to mask out elements written to the MemRef.
+    The `mask` type is an `i1` vector with a shape that matches how elements
+    are written to the MemRef.
----------------
MacDue wrote:

nit: I don't think the we need to say "SSA value" (it's already clear from that it's an operand and not an attribute). 

I also think this description could be simplified: 
```suggestion
    The provided `mask` is used to specify which elements of the tile slice will be stored.
```

This makes it clear it's masking the tile slice. The other details like it's writing to a `MemRef`, or that the mask is `i1`s should already be clear from context. :)



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


More information about the Mlir-commits mailing list