From 9cb10213b44a2b6414cf71a821de9b9c445dbc58 Mon Sep 17 00:00:00 2001
From: thibault allenet <thibault.allenet@cea.fr>
Date: Fri, 13 Dec 2024 15:21:38 +0000
Subject: [PATCH] Update export registry with new tensor entries
---
aidge_export_arm_cortexm/export_registry.py | 18 ++++++++++++++++++
1 file changed, 18 insertions(+)
diff --git a/aidge_export_arm_cortexm/export_registry.py b/aidge_export_arm_cortexm/export_registry.py
index 46c0653..a6e60b3 100644
--- a/aidge_export_arm_cortexm/export_registry.py
+++ b/aidge_export_arm_cortexm/export_registry.py
@@ -1,13 +1,31 @@
from aidge_core.export_utils import ExportLib
+from aidge_export_arm_cortexm.utils import ROOT
class ExportLibAidgeARM(ExportLib):
_name="aidge_arm"
+ def __init__(self, operator):
+ super(ExportLibAidgeARM, self).__init__(operator)
+ self.forward_template = str(ROOT / "_Aidge_Arm" / "templates" / "forward_call" / "forward.jinja")
+
# TODO ugly fix for Tensor registration issue...
import aidge_core
aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.float32],
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.float32]))
+aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.int32],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.int32]))
+aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.int8],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.int8]))
+aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.int4],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.int4]))
+aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.uint4],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.uint4]))
+aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.dual_int4],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.dual_int4]))
+aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.dual_uint4],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.dual_uint4]))
class ExportLibCMSISNN(ExportLib):
_name="export_cmsisnn"
+
--
GitLab