[Mlir-commits] [mlir] [MLIR][XeGPU] Skip control flow ops in recoverTemporaryLayout (PR #180771)

[Nishant Patel]

https://github.com/nbpatel created https://github.com/llvm/llvm-project/pull/180771

None

>From 6c9b83257d80fe08ba16ce4b2ec01b4141795223 Mon Sep 17 00:00:00 2001
From: nbpatel <nishant.b.patel at intel.com>
Date: Tue, 10 Feb 2026 16:30:15 +0000
Subject: [PATCH] Skip control flow ops in recoverTemporaryLayout

---
 mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp b/mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp
index a4e47fca96d34..291a39bca9938 100644
--- a/mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp
+++ b/mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp
@@ -91,7 +91,7 @@ bool xegpu::recoverTemporaryLayouts(Operation *rootOp) {
         continue;
       // Skip block arguments since they don't have defining ops to attach
       // layout attributes to
-      if (isa<BlockArgument>(operand.get()))
+      if (isa<BlockArgument>(operand.get()) || isa<LoopLikeOpInterface>(op))
         continue;
       auto layout = xegpu::getDistributeLayoutAttr(operand.get());
       if (!layout) {