11// test-onnx-to-mlir.cpp
22
33#include " mlir/Dialect/Func/IR/FuncOps.h"
4- #include " src/Dialect/ONNX/ONNXDialect.hpp"
5- #include " src/Builder/FrontendDialectTransformer.hpp"
6- #include " mlir/IR/MLIRContext.h"
74#include " mlir/IR/BuiltinOps.h"
85#include " mlir/IR/DialectRegistry.h"
6+ #include " mlir/IR/MLIRContext.h"
7+
8+ #include " src/Builder/FrontendDialectTransformer.hpp"
9+ #include " src/Dialect/ONNX/ONNXDialect.hpp"
910#include " llvm/Support/InitLLVM.h"
10- #include " llvm/Support/CommandLine .h"
11+ #include " llvm/Support/raw_ostream .h"
1112
1213int main (int argc, char **argv) {
1314 llvm::InitLLVM y (argc, argv);
@@ -18,7 +19,7 @@ int main(int argc, char **argv) {
1819 }
1920
2021 mlir::MLIRContext context;
21-
22+
2223 // Register the ONNX dialect.
2324 mlir::DialectRegistry registry;
2425 registry.insert <mlir::func::FuncDialect>();
@@ -30,17 +31,17 @@ int main(int argc, char **argv) {
3031 mlir::OwningOpRef<mlir::ModuleOp> module ;
3132 std::string errorMessage;
3233 onnx_mlir::ImportOptions options;
33-
34+
3435 int result = onnx_mlir::ImportFrontendModelFile (
3536 argv[1 ], context, module , &errorMessage, options);
36-
37+
3738 if (result != 0 ) {
3839 llvm::errs () << " Failed to import model: " << errorMessage << " \n " ;
3940 return 1 ;
4041 }
4142
4243 llvm::outs () << " Successfully imported ONNX model!\n " ;
4344 module ->print (llvm::outs ());
44-
45+
4546 return 0 ;
4647}
0 commit comments