[Triv2] Fix model's internal id handling in triv2

This patch fixes model's internal id handling in triv2.

Signed-off-by: Dongju Chae <dongju.chae@samsung.com>

diff --git a/packaging/npu-engine.spec b/packaging/npu-engine.spec
index a134efe..dffacdb 100644 (file)
--- a/packaging/npu-engine.spec
+++ b/packaging/npu-engine.spec
@@ -120,8 +120,7 @@ DESTDIR=%{buildroot} ninja install -C build %{?_smp_mflags}
 
   ./apptests/apptest_example_visa sample_models_v1 > /dev/null
   ./apptests/apptest_example_visa sample_models_v2 > /dev/null
-  # TODO: Let's make it work
-  # ./apptests/apptest_example_visa_triv2 sample_models_v3 > /dev/null
+  ./apptests/apptest_example_visa_triv2 sample_models_v3 > /dev/null
 
   rm -rf sample_models_*
 

diff --git a/src/core/ne-handler.cc b/src/core/ne-handler.cc
index b378efa..53227c2 100644 (file)
--- a/src/core/ne-handler.cc
+++ b/src/core/ne-handler.cc
@@ -1335,6 +1335,7 @@ TrinityVision2::setModel (const generic_buffer *model_buf, Model ** model_ptr)
   if (status != 0)
     goto delete_exit;
 
+  model->setInternalID(config.id);
   *model_ptr = model;
   return status;