[Mlir-commits] [mlir] [MLIR][XeGPU][TransformOps] set_op_layout_attr supports setting anchor layout (PR #172542)

[Tuomas Kärnä]

================
@@ -347,11 +349,24 @@ transform::SetOpLayoutAttrOp::apply(transform::TransformRewriter &rewriter,
         getContext(), layout, DenseI64ArrayAttr::get(getContext(), sliceDims));
   }
 
-  // Set layout attribute for the op result or operand
-  if (resultTarget)
+  // Set layout attribute
+  if (resultTarget) {
+    // for the op result
     xegpu::setDistributeLayoutAttr(target->getResult(index), layout);
-  else
+  } else if (operandTarget) {
+    // or operand
     xegpu::setDistributeLayoutAttr(target->getOpOperand(index), layout);
+  } else {
+    // or anchor layout.
----------------
tkarna wrote:

Yes, the transform ops can be cleaned up a lot now. I'll defer this to another later stage, however, when we have better workload coverage and know what's actually needed.

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