diff options
Diffstat (limited to 'standalone-opt/standalone-opt.cpp')
-rw-r--r-- | standalone-opt/standalone-opt.cpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/standalone-opt/standalone-opt.cpp b/standalone-opt/standalone-opt.cpp index 97a996a..e84628f 100644 --- a/standalone-opt/standalone-opt.cpp +++ b/standalone-opt/standalone-opt.cpp @@ -6,6 +6,7 @@ // //===----------------------------------------------------------------------===// +#include "mlir/Dialect/Arithmetic/IR/Arithmetic.h" #include "mlir/IR/Dialect.h" #include "mlir/IR/MLIRContext.h" #include "mlir/InitAllDialects.h" @@ -13,7 +14,7 @@ #include "mlir/Pass/Pass.h" #include "mlir/Pass/PassManager.h" #include "mlir/Support/FileUtilities.h" -#include "mlir/Support/MlirOptMain.h" +#include "mlir/Tools/mlir-opt/MlirOptMain.h" #include "llvm/Support/CommandLine.h" #include "llvm/Support/InitLLVM.h" #include "llvm/Support/SourceMgr.h" @@ -26,8 +27,8 @@ int main(int argc, char **argv) { // TODO: Register standalone passes here. mlir::DialectRegistry registry; - registry.insert<mlir::standalone::StandaloneDialect>(); - registry.insert<mlir::StandardOpsDialect>(); + registry.insert<mlir::standalone::StandaloneDialect, + mlir::arith::ArithmeticDialect, mlir::func::FuncDialect>(); // Add the following to include *all* MLIR Core dialects, or selectively // include what you need like above. You only need to register dialects that // will be *parsed* by the tool, not the one generated |