[Mlir-commits] [mlir] 798e4bf - [mlir] Fix integration tests failures introduced in D108505

[Vladislav Vinogradov]

Author: Vladislav Vinogradov
Date: 2021-09-20T11:48:24+03:00
New Revision: 798e4bfbeda824551fa89a388969baa2abbc2411

URL: https://github.com/llvm/llvm-project/commit/798e4bfbeda824551fa89a388969baa2abbc2411
DIFF: https://github.com/llvm/llvm-project/commit/798e4bfbeda824551fa89a388969baa2abbc2411.diff

LOG: [mlir] Fix integration tests failures introduced in D108505

Added: 
    

Modified: 
    mlir/lib/Dialect/Linalg/Transforms/Bufferize.cpp
    mlir/lib/Dialect/Tensor/Transforms/Bufferize.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Linalg/Transforms/Bufferize.cpp b/mlir/lib/Dialect/Linalg/Transforms/Bufferize.cpp
index 60ccd0bd787d3..79a111f73949f 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/Bufferize.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/Bufferize.cpp
@@ -333,8 +333,9 @@ struct LinalgBufferizePass : public LinalgBufferizeBase<LinalgBufferizePass> {
       return typeConverter.isLegal(op);
     };
     target.addDynamicallyLegalDialect<linalg::LinalgDialect>(isLegalOperation);
-    target.addDynamicallyLegalOp<ConstantOp, vector::TransferReadOp,
-                                 vector::TransferWriteOp>(isLegalOperation);
+    target
+        .addDynamicallyLegalOp<vector::TransferReadOp, vector::TransferWriteOp>(
+            isLegalOperation);
 
     RewritePatternSet patterns(&context);
     populateLinalgBufferizePatterns(typeConverter, patterns);

diff  --git a/mlir/lib/Dialect/Tensor/Transforms/Bufferize.cpp b/mlir/lib/Dialect/Tensor/Transforms/Bufferize.cpp
index f5f7b0f5faf19..e8c38656ec21d 100644
--- a/mlir/lib/Dialect/Tensor/Transforms/Bufferize.cpp
+++ b/mlir/lib/Dialect/Tensor/Transforms/Bufferize.cpp
@@ -175,6 +175,7 @@ struct TensorBufferizePass : public TensorBufferizeBase<TensorBufferizePass> {
     target.addLegalDialect<memref::MemRefDialect>();
     target.addDynamicallyLegalDialect<StandardOpsDialect>(
         [&](Operation *op) { return typeConverter.isLegal(op); });
+    target.addLegalOp<CallOp>();
     target.addLegalOp<ReturnOp>();
     target.addLegalDialect<scf::SCFDialect>();