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

[llvmlistbot at llvm.org]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-mlir

Author: Nishant Patel (nbpatel)

<details>
<summary>Changes</summary>



---
Full diff: https://github.com/llvm/llvm-project/pull/180771.diff


1 Files Affected:

- (modified) mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp (+1-1) 


``````````diff
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) {

``````````

</details>


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