[Mlir-commits] [mlir] 8102ebf - [MLIR][XeGPU] Fixing PR179016 minor issues (#180295)

[llvmlistbot at llvm.org]

Author: Jianhui Li
Date: 2026-02-06T14:51:40-08:00
New Revision: 8102ebf6a3c8c5d2665aa11d0fb5aa264ffe1b04

URL: https://github.com/llvm/llvm-project/commit/8102ebf6a3c8c5d2665aa11d0fb5aa264ffe1b04
DIFF: https://github.com/llvm/llvm-project/commit/8102ebf6a3c8c5d2665aa11d0fb5aa264ffe1b04.diff

LOG: [MLIR][XeGPU] Fixing PR179016 minor issues (#180295)

Fix two issues brough by PR179016: 
1. unused variable if build the option with
"DLLVM_ENABLE_ASSERTIONS=OFF"
2. Recover modification to recoverTemporaryLayouts() brought by
PR176737. Unintentionally lost during the merging process.

Added: 
    

Modified: 
    mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp b/mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp
index 6e908cbf8c1e0..738ff3f8aa915 100644
--- a/mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp
+++ b/mlir/lib/Dialect/XeGPU/Transforms/XeGPULayoutImpl.cpp
@@ -91,9 +91,9 @@ bool xegpu::recoverTemporaryLayouts(Operation *rootOp) {
         continue;
       auto layout = xegpu::getDistributeLayoutAttr(operand.get());
       if (!layout) {
-        op->emitError("Could not find layout attribute for operand ")
+        op->emitWarning("Could not find layout attribute for operand ")
             << operand.getOperandNumber() << " of operation " << op->getName();
-        return WalkResult::interrupt();
+        continue;
       }
       xegpu::setDistributeLayoutAttr(operand, layout);
     }
@@ -167,8 +167,8 @@ xegpu::inferMultiReductionSourceLayout(xegpu::DistributeLayoutAttr resLayout,
          "reduction result layout must be slice layout");
 
   xegpu::SliceAttr sliceLayout = dyn_cast<xegpu::SliceAttr>(resLayout);
-  auto sliceDims = sliceLayout.getDims().asArrayRef();
-  assert(reduceDims == sliceDims &&
+
+  assert((reduceDims == sliceLayout.getDims().asArrayRef()) &&
          "reduction dims must match with slice dims");
 
   return sliceLayout.getParent();