[Mlir-commits] [mlir] [mlir][xegpu] Remove chunk_size attribute; infer from types (PR #205122)

[Jianhui Li]

================
@@ -237,15 +251,17 @@ xegpu::getDistributeLayoutAttr(const OpOperand &opr) {
 
     if (isa<xegpu::StoreScatterOp>(op)) {
       xegpu::StoreScatterOp store(op);
-      int chunkSize = store.getChunkSize().value_or(1);
+      int chunkSize = xegpu::getGatherScatterPayloadChunk(store.getValueType(),
+                                                          store.getMaskType());
       if (layout && idx >= 2 && chunkSize > 1)
         return layout.dropDims(llvm::to_vector(
----------------
Jianhui-Li wrote:

remove these special processing. the mask/offset have the same shape as data now. 

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