From 3bc30f544481bc3f4bdcd46742c4df6af4154827 Mon Sep 17 00:00:00 2001
From: Jiyoung Yun <jy910.yun@samsung.com>
Date: Fri, 9 Aug 2024 10:45:51 +0900
Subject: [PATCH] [onert] Store other parameter offset in CheckpointExporter

ONE-DCO-1.0-Signed-off-by: Jiyoung Yun <jy910.yun@samsung.com>
---
 runtime/onert/core/src/exporter/CheckpointExporter.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/runtime/onert/core/src/exporter/CheckpointExporter.cc b/runtime/onert/core/src/exporter/CheckpointExporter.cc
index f88293d9ed5..dbcfe49a4a1 100644
--- 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