maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir] Fix integration tests failures introduced in D108505

This commit is contained in:
Vladislav Vinogradov 2021-09-20 11:48:24 +03:00
parent 92c9b28347
commit 798e4bfbed
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mlir/lib/Dialect/Linalg/Transforms/Bufferize.cpp
View File

@ -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);

1
mlir/lib/Dialect/Tensor/Transforms/Bufferize.cpp
View File

@ -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>();