From 6558fa641b2b762f710c02448c887c40e60b1d18 Mon Sep 17 00:00:00 2001 From: Håvard Pettersen Date: Wed, 23 Jun 2021 13:53:28 +0000 Subject: dry run onnx models on setup --- eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'eval/src/apps/analyze_onnx_model') 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; + } +} -- cgit v1.2.3