[clang-tools-extra] [mlir] [MLIR][XeGPU] Refactor Layout access interface (PR #172125)

[Tuomas Kärnä via cfe-commits]

================
@@ -1508,19 +1517,20 @@ void XeGPUWgToSgDistributePass::runOnOperation() {
                                vector::ConstantMaskOp, vector::CreateMaskOp>(
       [=](Operation *op) -> bool {
         // Check for either a SliceAttr or LayoutAttr on the result.
-        auto layout = xegpu::getDistributeLayoutAttr(op->getResult(0));
+        auto layout =
+            xegpu::getTempLayout(dyn_cast<OpResult>(op->getResult(0)));
         return isLegal(layout);
       });
 
   target.addDynamicallyLegalOp<xegpu::LoadGatherOp>(
       [=](xegpu::LoadGatherOp op) -> bool {
-        auto layout = xegpu::getDistributeLayoutAttr(op.getResult());
+        auto layout = op.getLayoutAttr();
----------------
tkarna wrote:

getAnchorLayout

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