1
1
// test-onnx-to-mlir.cpp
2
2
3
3
#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"
7
4
#include " mlir/IR/BuiltinOps.h"
8
5
#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"
9
10
#include " llvm/Support/InitLLVM.h"
10
- #include " llvm/Support/CommandLine .h"
11
+ #include " llvm/Support/raw_ostream .h"
11
12
12
13
int main (int argc, char **argv) {
13
14
llvm::InitLLVM y (argc, argv);
@@ -18,7 +19,7 @@ int main(int argc, char **argv) {
18
19
}
19
20
20
21
mlir::MLIRContext context;
21
-
22
+
22
23
// Register the ONNX dialect.
23
24
mlir::DialectRegistry registry;
24
25
registry.insert <mlir::func::FuncDialect>();
@@ -30,17 +31,17 @@ int main(int argc, char **argv) {
30
31
mlir::OwningOpRef<mlir::ModuleOp> module ;
31
32
std::string errorMessage;
32
33
onnx_mlir::ImportOptions options;
33
-
34
+
34
35
int result = onnx_mlir::ImportFrontendModelFile (
35
36
argv[1 ], context, module , &errorMessage, options);
36
-
37
+
37
38
if (result != 0 ) {
38
39
llvm::errs () << " Failed to import model: " << errorMessage << " \n " ;
39
40
return 1 ;
40
41
}
41
42
42
43
llvm::outs () << " Successfully imported ONNX model!\n " ;
43
44
module ->print (llvm::outs ());
44
-
45
+
45
46
return 0 ;
46
47
}
0 commit comments