diff options
author | Mehdi Amini <joker.eph@gmail.com> | 2022-01-01 21:19:31 +0000 |
---|---|---|
committer | Mehdi Amini <joker.eph@gmail.com> | 2022-01-01 21:19:31 +0000 |
commit | 104a827ea6de0cbe0f5faef4407552ede31d165c (patch) | |
tree | 4bbfa8ef97cd07195ae4a2d17e7bb33fc8c4fb4f | |
parent | a978847e3aa4bdd64b242a31bc9037019581d1e3 (diff) | |
download | llvm-104a827ea6de0cbe0f5faef4407552ede31d165c.zip llvm-104a827ea6de0cbe0f5faef4407552ede31d165c.tar.gz llvm-104a827ea6de0cbe0f5faef4407552ede31d165c.tar.bz2 |
Move LinalgDetensorize pass option from .cpp file to the .td declaration (NFC)
-rw-r--r-- | mlir/include/mlir/Dialect/Linalg/Passes.td | 6 | ||||
-rw-r--r-- | mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp | 7 |
2 files changed, 6 insertions, 7 deletions
diff --git a/mlir/include/mlir/Dialect/Linalg/Passes.td b/mlir/include/mlir/Dialect/Linalg/Passes.td index 5bcc8cc..893487d 100644 --- a/mlir/include/mlir/Dialect/Linalg/Passes.td +++ b/mlir/include/mlir/Dialect/Linalg/Passes.td @@ -237,6 +237,12 @@ def LinalgDetensorize : Pass<"linalg-detensorize", ""> { run on others. This is because it performs specific legalization of the blocks that make up the body, which it assumes has a FunctionLike trait. }]; + let options = [ + Option<"aggressiveMode", "aggressive-mode", "bool", /*default=*/"false", + "Detensorize all ops that qualify for detensoring along with branch" + " operands and basic-block arguments."> + + ]; } def LinalgStrategyTileAndFusePass diff --git a/mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp b/mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp index 9be95a1..a20da7f 100644 --- a/mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp +++ b/mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp @@ -196,8 +196,6 @@ struct ExtractFromReshapeFromElements /// @see LinalgDetensorize in Linalg/Passes.td for more details. struct LinalgDetensorize : public LinalgDetensorizeBase<LinalgDetensorize> { LinalgDetensorize() = default; - LinalgDetensorize(const LinalgDetensorize &pass) - : LinalgDetensorizeBase<LinalgDetensorize>() {} class CostModel { public: @@ -600,11 +598,6 @@ struct LinalgDetensorize : public LinalgDetensorizeBase<LinalgDetensorize> { std::move(canonPatterns)))) signalPassFailure(); } - - Option<bool> aggressiveMode{ - *this, "aggressive-mode", - llvm::cl::desc("Detensorize all ops that qualify for detensoring along " - "with branch operands and basic-block arguments.")}; }; } // namespace |