[Mlir-commits] [mlir] 693c61b - [mlir] Enable loop peeling for "reduction" dimensions of tiled_loop.

[Alexander Belyaev]

Author: Alexander Belyaev
Date: 2021-10-01T13:07:57+02:00
New Revision: 693c61b2e05005b584c53e388da3f971e1ced747

URL: https://github.com/llvm/llvm-project/commit/693c61b2e05005b584c53e388da3f971e1ced747
DIFF: https://github.com/llvm/llvm-project/commit/693c61b2e05005b584c53e388da3f971e1ced747.diff

LOG: [mlir] Enable loop peeling for "reduction" dimensions of tiled_loop.

Differential Revision: https://reviews.llvm.org/D110919

Added: 
    

Modified: 
    mlir/lib/Dialect/Linalg/Transforms/Loops.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Linalg/Transforms/Loops.cpp b/mlir/lib/Dialect/Linalg/Transforms/Loops.cpp
index 07d9eef56b8e4..42105a9534719 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/Loops.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/Loops.cpp
@@ -725,9 +725,6 @@ LogicalResult mlir::linalg::peelAndCanonicalizeTiledLoop(RewriterBase &rewriter,
   int64_t numLoops = loopOp.iterator_types().size();
   if (idx < 0 || numLoops <= idx)
     return failure();
-  // Only parallel iterator supported.
-  if (!isParallelIterator(loopOp.iterator_types()[idx]))
-    return failure();
 
   Value ub = loopOp.upperBound()[idx];
   TiledLoopOp remainderLoop;