From ba833c320a9b207c4a540db7c64aa22933543e46 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Fri, 13 Sep 2024 14:06:18 +0000
Subject: [PATCH] Update meminfo default values.
---
aidge_core/export_utils/node_export.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/aidge_core/export_utils/node_export.py b/aidge_core/export_utils/node_export.py
index eeba09c92..1154e88f3 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -240,11 +240,11 @@ class ExportNode(ABC):
if "cont_size" in mem_info[idx]:
self.attributes["mem_info_cont_size"][idx] = mem_info[idx]["cont_size"]
else:
- self.attributes["mem_info_cont_size"][idx] = tensor.size()
+ self.attributes["mem_info_cont_size"][idx] = mem_info[idx]["size"]
if "cont_offset" in mem_info[idx]:
self.attributes["mem_info_cont_offset"][idx] = mem_info[idx]["cont_offset"]
else:
- self.attributes["mem_info_cont_offset"][idx] = 0
+ self.attributes["mem_info_cont_offset"][idx] = mem_info[idx]["offset"]
if "cont_offset" in mem_info[idx]:
self.attributes["mem_info_wrap_offset"][idx] = mem_info[idx]["wrap_offset"]
else:
--
GitLab