Fix setModel
authorJiho Chu <jiho.chu@samsung.com>
Fri, 19 Jul 2024 10:28:05 +0000 (19:28 +0900)
committerWook Song <wook16.song@samsung.com>
Tue, 18 Mar 2025 07:28:09 +0000 (16:28 +0900)
It fixes setModel for zero-size program model.
If program size is zeio, it skips setting program section, and
continue with weight.

Signed-off-by: Jiho Chu <jiho.chu@samsung.com>
src/core/ne-handler.cc

index 5b48f04c04a45015300c8212491c14c07fb0cfdd..a3042484d4b1af7586bb2cea7c3341252157654b 100644 (file)
@@ -1081,6 +1081,8 @@ TrinityVision2::setModel (const generic_buffer *model_buf, const generic_buffer
       logerr (TAG, "Failed to extract generic buffer: %d\n", status);
       goto delete_exit;
     }
+  } else {
+         goto set_weight;
   }
 
   /** register this model to the driver */
@@ -1128,6 +1130,7 @@ TrinityVision2::setModel (const generic_buffer *model_buf, const generic_buffer
 
   model->setInternalID (config.id);
 
+set_weight:
   if (weight) {
     hwmem_weight = new HWmem (new HWmemDevice);