Exception handling on tflite_run (#4531)

- Add noexcept keyword for Args constructor
- Add try-catch for external tflite interpreter setting

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>

3 files changed, 16 insertions, 8 deletions

diff --git a/tests/tools/tflite_run/src/args.cc b/tests/tools/tflite_run/src/args.cc
index 713a0a9d2..eed287bd9 100644
--- a/tests/tools/tflite_run/src/args.cc
+++ b/tests/tools/tflite_run/src/args.cc
@@ -23,7 +23,7 @@
 namespace TFLiteRun
 {
 
-Args::Args(const int argc, char **argv)
+Args::Args(const int argc, char **argv) noexcept
 {
   Initialize();
   Parse(argc, argv);
diff --git a/tests/tools/tflite_run/src/args.h b/tests/tools/tflite_run/src/args.h
index 5561544eb..a2d53391c 100644
--- a/tests/tools/tflite_run/src/args.h
+++ b/tests/tools/tflite_run/src/args.h
@@ -28,7 +28,7 @@ namespace TFLiteRun
 class Args
 {
 public:
-  Args(const int argc, char **argv);
+  Args(const int argc, char **argv) noexcept;
   void print(void);
 
   const std::string &getTFLiteFilename(void) const { return _tflite_filename; }
diff --git a/tests/tools/tflite_run/src/tflite_run.cc b/tests/tools/tflite_run/src/tflite_run.cc
index 5be6909e5..96f2c4155 100644
--- a/tests/tools/tflite_run/src/tflite_run.cc
+++ b/tests/tools/tflite_run/src/tflite_run.cc
@@ -65,15 +65,23 @@ int main(const int argc, char **argv)
   std::chrono::milliseconds t_prepare(0);
   std::chrono::milliseconds t_invoke(0);
 
-  nnfw::misc::benchmark::measure(t_prepare) << [&](void) {
-    BuiltinOpResolver resolver;
+  try
+  {
+    nnfw::misc::benchmark::measure(t_prepare) << [&](void) {
+      BuiltinOpResolver resolver;
 
-    InterpreterBuilder builder(*model, resolver);
+      InterpreterBuilder builder(*model, resolver);
 
-    TFLITE_ENSURE(builder(&interpreter))
+      TFLITE_ENSURE(builder(&interpreter))
 
-    interpreter->SetNumThreads(1);
-  };
+      interpreter->SetNumThreads(1);
+    };
+  }
+  catch (const std::exception &e)
+  {
+    std::cerr << e.what() << '\n';
+    return 1;
+  }
 
   std::shared_ptr<nnfw::tflite::Session> sess;