File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -58,13 +58,11 @@ void populateVectorPasses(mlir::PassManager &pm) {
58
58
// scf + arith + math + vector + memref + linalg.brgemm
59
59
void populateBufferizationPasses (mlir::PassManager &pm) {
60
60
bufferization::OneShotBufferizationOptions options;
61
+ options.bufferizeFunctionBoundaries = true ;
62
+ options.setFunctionBoundaryTypeConversion (
63
+ bufferization::LayoutMapOption::IdentityLayoutMap);
61
64
pm.addPass (bufferization::createOneShotBufferizePass (options));
62
65
pm.addPass (createCSEPass ());
63
- pm.addPass (mlir::func::createFuncBufferizePass ());
64
- pm.addNestedPass <func::FuncOp>(
65
- bufferization::createBufferizationBufferizePass ());
66
- pm.addNestedPass <func::FuncOp>(
67
- bufferization::createFinalizingBufferizePass ());
68
66
bufferization::BufferResultsToOutParamsOpts opt{};
69
67
opt.hoistStaticAllocs = true ;
70
68
pm.addPass (bufferization::createBufferResultsToOutParamsPass (opt));
You can’t perform that action at this time.
0 commit comments