aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMehdi Amini <joker.eph@gmail.com>2022-01-01 21:19:31 +0000
committerMehdi Amini <joker.eph@gmail.com>2022-01-01 21:19:31 +0000
commit104a827ea6de0cbe0f5faef4407552ede31d165c (patch)
tree4bbfa8ef97cd07195ae4a2d17e7bb33fc8c4fb4f
parenta978847e3aa4bdd64b242a31bc9037019581d1e3 (diff)
downloadllvm-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.td6
-rw-r--r--mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp7
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