xref: /llvm-project/mlir/examples/transform/Ch4/transform-opt/transform-opt.cpp (revision b33b91a21788d439f49d6db4e7224c20f740f1a7)
1 //===-- transform-opt.cpp - Transform dialect tutorial entry point --------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This is the top-level file for the Transform dialect tutorial chapter 4.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "MyExtension.h"
14 
15 #include "mlir/Dialect/Transform/Transforms/Passes.h"
16 #include "mlir/IR/DialectRegistry.h"
17 #include "mlir/IR/MLIRContext.h"
18 #include "mlir/InitAllDialects.h"
19 #include "mlir/InitAllExtensions.h"
20 #include "mlir/Tools/mlir-opt/MlirOptMain.h"
21 #include "mlir/Transforms/Passes.h"
22 #include <cstdlib>
23 
main(int argc,char ** argv)24 int main(int argc, char **argv) {
25   // Register all "core" dialects and our transform dialect extension.
26   mlir::DialectRegistry registry;
27   mlir::registerAllDialects(registry);
28   mlir::registerAllExtensions(registry);
29   registerMyExtension(registry);
30 
31   // Register a handful of cleanup passes that we can run to make the output IR
32   // look nicer.
33   mlir::registerCanonicalizerPass();
34   mlir::registerCSEPass();
35   mlir::registerSymbolDCEPass();
36   mlir::transform::registerInterpreterPass();
37 
38   // Delegate to the MLIR utility for parsing and pass management.
39   return mlir::MlirOptMain(argc, argv, "transform-opt-ch4", registry)
40                  .succeeded()
41              ? EXIT_SUCCESS
42              : EXIT_FAILURE;
43 }
44