[onert] Store other parameter offset in CheckpointExporter

ONE-DCO-1.0-Signed-off-by: Jiyoung Yun <[email protected]>
Samsung · Aug 9, 2024 · 3bc30f5 · 3bc30f5
1 parent 51231c7
commit 3bc30f5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/runtime/onert/core/src/exporter/CheckpointExporter.cc b/runtime/onert/core/src/exporter/CheckpointExporter.cc
@@ -261,6 +261,11 @@ void CheckpointExporter::setAdamOptimizerData(const std::unique_ptr<onert::exec:
 
   // Write Adam V offset
   memcpy(ptr, &v_offset, sizeof(v_offset));
+  ptr += sizeof(v_offset);
+
+  // Write other param offset
+  uint32_t o_offset = v_offset + v_offset_size + v_total_size;
+  memcpy(ptr, &o_offset, sizeof(o_offset));
 }
 
 } // namespace exporter