diff options
Diffstat (limited to 'eval')
-rw-r--r-- | eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp b/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp index 3f56610dcaa..ce7070d6b2b 100644 --- a/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp +++ b/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp @@ -172,7 +172,7 @@ int usage(const char *self) { return 1; } -int main(int argc, char **argv) { +int my_main(int argc, char **argv) { if (argc < 2) { return usage(argv[0]); } @@ -206,3 +206,12 @@ int main(int argc, char **argv) { fprintf(stderr, "estimated model evaluation time: %g ms\n", min_time_s * 1000.0); return 0; } + +int main(int argc, char **argv) { + try { + return my_main(argc, argv); + } catch (const std::exception &ex) { + fprintf(stderr, "got exception: %s\n", ex.what()); + return 2; + } +} |