dry run onnx models on setup

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;
+    }
+}