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

Adds a flag to optionally disable tosa decompositions 

Reviewed By: rsuderman

Differential Revision: https://reviews.llvm.org/D120338
This commit is contained in:
not-jenni 2022-02-28 15:30:13 -08:00 committed by Rob Suderman
parent 19d8642633
commit fd2550d80c
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mlir/include/mlir/Conversion/TosaToLinalg/TosaToLinalg.h
View File

@ -24,8 +24,10 @@ std::unique_ptr<Pass> createTosaToLinalgNamed();
/// Populates passes to convert from TOSA to Linalg on buffers. At the end of
/// the pass, the function will only contain linalg ops or standard ops if the
/// pipeline succeeds.
void addTosaToLinalgPasses(OpPassManager &pm);
/// pipeline succeeds. The option to disable decompositions is available for
/// benchmarking performance improvements from the canonicalizations.
void addTosaToLinalgPasses(OpPassManager &pm,
bool disableTosaDecompositions = false);
/// Populates conversion passes from TOSA dialect to Linalg dialect.
void populateTosaToLinalgConversionPatterns(RewritePatternSet *patterns);

6
mlir/lib/Conversion/TosaToLinalg/TosaToLinalgPass.cpp
View File

@ -67,9 +67,11 @@ std::unique_ptr<Pass> mlir::tosa::createTosaToLinalg() {
return std::make_unique<TosaToLinalg>();
}
void mlir::tosa::addTosaToLinalgPasses(OpPassManager &pm) {
void mlir::tosa::addTosaToLinalgPasses(OpPassManager &pm,
bool disableTosaDecompositions) {
// Optional decompositions are designed to benefit linalg.
pm.addNestedPass<FuncOp>(mlir::tosa::createTosaOptionalDecompositions());
if (!disableTosaDecompositions)
pm.addNestedPass<FuncOp>(mlir::tosa::createTosaOptionalDecompositions());
pm.addNestedPass<FuncOp>(mlir::createCanonicalizerPass());
pm.addNestedPass<FuncOp>(tosa::createTosaMakeBroadcastablePass());