Add selected MLIR options (#213)

* Reorganize main function.

* Follow review comments.

* Emit constants are globals in Krnl and LLVM dialects.

* Add selected MLIR passes.

* Refactor.

* Fix descriptions.

* Refactor.

MLIR.cmake

@@ -152,6 +152,7 @@ find_mlir_lib(MLIRLinalgAnalysis)
 find_mlir_lib(MLIRLinalgTransforms)
 find_mlir_lib(MLIRLinalgUtils)
 find_mlir_lib(MLIRSCF)
+find_mlir_lib(MLIRSCFTransforms)
 find_mlir_lib(MLIRLLVMIRTransforms)
 find_mlir_lib(MLIRMlirOptMain)
 find_mlir_lib(MLIRParser)
@@ -171,6 +172,8 @@ find_mlir_lib(MLIRTargetLLVMIR)
 find_mlir_lib(MLIRTransformUtils)
 find_mlir_lib(MLIRTranslation)
 find_mlir_lib(MLIRVector)
+find_mlir_lib(MLIRVectorToLLVM)
+find_mlir_lib(MLIRVectorToSCF)
 find_mlir_lib(MLIRMlirOptMain)
 find_mlir_lib(MLIRAffineEDSC)
 find_mlir_lib(MLIRLinalgEDSC)
@@ -203,6 +206,8 @@ set(MLIRLibs
         ${MLIRTransforms}
         ${MLIRSCFToStandard}
         ${MLIRVector}
+        ${MLIRVectorToLLVM}
+        ${MLIRVectorToSCF}
         ${MLIRSCF}
         ${MLIRIR}
         ${MLIRLLVMIR}
@@ -226,6 +231,7 @@ set(MLIRLibs
         ${MLIRLLVMIRTransforms}        
         ${MLIRSCFToStandard}
         ${MLIRSCF}
+        ${MLIRSCFTransforms}
         ${MLIRLoopAnalysis}
         ${MLIRLoopLikeInterface}
         ${MLIROpenMP}

src/InitMLIRPasses.hpp

@@ -0,0 +1,25 @@
+#include "mlir/Pass/Pass.h"
+
+namespace onnx_mlir {
+
+void initMLIRPasses() {
+  mlir::registerPass("convert-vector-to-scf", "Convert vector to SCF.",
+      []() -> std::unique_ptr<mlir::Pass> {
+        return mlir::createConvertVectorToSCFPass();
+      });
+  mlir::registerPass("lower-affine",
+      "Lower Affine Dialect to Standard Dialect.",
+      []() -> std::unique_ptr<mlir::Pass> {
+        return mlir::createLowerAffinePass();
+      });
+  mlir::registerPass("convert-scf-to-std", "Lower SCF to Standard Dialect.",
+      []() -> std::unique_ptr<mlir::Pass> {
+        return mlir::createLowerToCFGPass();
+      });
+  mlir::registerPass("convert-vector-to-llvm",
+      "Lower Vector Dialect to LLVM IR Dialect.",
+      []() -> std::unique_ptr<mlir::Pass> {
+        return mlir::createConvertVectorToLLVMPass();
+      });
+}
+} // namespace onnx_mlir

src/Tool/ONNXMLIROpt/ONNXMLIROpt.cpp

@@ -24,6 +24,7 @@
 
 #include "src/Dialect/Krnl/KrnlOps.hpp"
 #include "src/Dialect/ONNX/ONNXOps.hpp"
+#include "src/InitMLIRPasses.hpp"
 #include "src/InitOMPasses.hpp"
 #include "src/Pass/Passes.hpp"
 
@@ -64,6 +65,7 @@ int main(int argc, char **argv) {
   mlir::registerDialect<mlir::LLVM::LLVMDialect>();
   mlir::registerDialect<mlir::scf::SCFDialect>();
   mlir::registerDialect<mlir::StandardOpsDialect>();
+  mlir::registerDialect<mlir::vector::VectorDialect>();
 
   // Register transformation passes.
 #define GEN_PASS_REGISTRATION
@@ -75,11 +77,15 @@ int main(int argc, char **argv) {
 #define GEN_PASS_REGISTRATION
 #include "mlir/Dialect/Linalg/Passes.h.inc"
 
+#define GEN_PASS_REGISTRATION
+#include "mlir/Dialect/SCF/Passes.h.inc"
+
   llvm::InitLLVM y(argc, argv);
 
   mlir::registerDialect<mlir::ONNXOpsDialect>();
   mlir::registerDialect<mlir::KrnlOpsDialect>();
   initOMPasses();
+  initMLIRPasses();
 
   mlir::registerAsmPrinterCLOptions();
   mlir::registerMLIRContextCLOptions();