[nnpkg_run] add load option (#6688)

This patch adds `load` option.
It does not load input files yet.

Signed-off-by: Sanggyu Lee <sg5.lee@samsung.com>

diff --git a/tests/tools/nnpackage_run/src/args.cc b/tests/tools/nnpackage_run/src/args.cc
index bcdbb91..ef787c8 100644 (file)
--- a/tests/tools/nnpackage_run/src/args.cc
+++ b/tests/tools/nnpackage_run/src/args.cc
@@ -37,7 +37,8 @@ void Args::Initialize(void)
   general.add_options()
     ("help,h", "Display available options")
     ("nnpackage", po::value<std::string>()->required())
-    ("dump,d", po::value<std::string>()->default_value(""), "Output filename");
+    ("dump,d", po::value<std::string>()->default_value(""), "Output filename")
+    ("load,l", po::value<std::string>()->default_value(""), "Input filename");
   // clang-format on
 
   _options.add(general);
@@ -76,6 +77,11 @@ void Args::Parse(const int argc, char **argv)
     _dump_filename = vm["dump"].as<std::string>();
   }
 
+  if (vm.count("load"))
+  {
+    _load_filename = vm["load"].as<std::string>();
+  }
+
   if (vm.count("nnpackage"))
   {
     _package_filename = vm["nnpackage"].as<std::string>();

diff --git a/tests/tools/nnpackage_run/src/args.h b/tests/tools/nnpackage_run/src/args.h
index 4268700..d064d77 100644 (file)
--- a/tests/tools/nnpackage_run/src/args.h
+++ b/tests/tools/nnpackage_run/src/args.h
@@ -33,6 +33,7 @@ public:
 
   const std::string &getPackageFilename(void) const { return _package_filename; }
   const std::string &getDumpFilename(void) const { return _dump_filename; }
+  const std::string &getLoadFilename(void) const { return _load_filename; }
 
 private:
   void Initialize();
@@ -44,6 +45,7 @@ private:
 
   std::string _package_filename;
   std::string _dump_filename;
+  std::string _load_filename;
 };
 
 } // end of namespace NNPackageRun