[Mlir-commits] [mlir] 3b42686 - [mlir][bufferize] Do not assert destination passing style for non-bufferizable ops

[Matthias Springer]

Author: Matthias Springer
Date: 2022-02-22T17:42:02+09:00
New Revision: 3b4268686965c6479dd29471287f2d4bc9cfcfaa

URL: https://github.com/llvm/llvm-project/commit/3b4268686965c6479dd29471287f2d4bc9cfcfaa
DIFF: https://github.com/llvm/llvm-project/commit/3b4268686965c6479dd29471287f2d4bc9cfcfaa.diff

LOG: [mlir][bufferize] Do not assert destination passing style for non-bufferizable ops

Added: 
    

Modified: 
    mlir/lib/Dialect/Bufferization/Transforms/OneShotAnalysis.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Bufferization/Transforms/OneShotAnalysis.cpp b/mlir/lib/Dialect/Bufferization/Transforms/OneShotAnalysis.cpp
index 3e643aae5745..340f28526b2d 100644
--- a/mlir/lib/Dialect/Bufferization/Transforms/OneShotAnalysis.cpp
+++ b/mlir/lib/Dialect/Bufferization/Transforms/OneShotAnalysis.cpp
@@ -707,7 +707,8 @@ assertDestinationPassingStyle(Operation *op, BufferizationState &state,
   LogicalResult status = success();
   DominanceInfo domInfo(op);
   op->walk([&](Operation *returnOp) {
-    if (!isRegionReturnLike(returnOp))
+    if (!isRegionReturnLike(returnOp) ||
+        !state.getOptions().isOpAllowed(returnOp))
       return WalkResult::advance();
 
     for (OpOperand &returnValOperand : returnOp->getOpOperands()) {