From 037e6205eb02a92a8662201369aaed9a039526ac Mon Sep 17 00:00:00 2001
From: thibault allenet <thibault.allenet@cea.fr>
Date: Fri, 13 Dec 2024 16:05:01 +0000
Subject: [PATCH] Update the export registry for tensor with new aidge
datatypes
---
aidge_core/export_utils/export_registry.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/aidge_core/export_utils/export_registry.py b/aidge_core/export_utils/export_registry.py
index e5b6b2098..8927ae516 100644
--- a/aidge_core/export_utils/export_registry.py
+++ b/aidge_core/export_utils/export_registry.py
@@ -80,6 +80,14 @@ class ExportLib(aidge_core.OperatorImpl):
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.uint32]))
aidge_core.register_Tensor([self._name, aidge_core.dtype.uint64],
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.uint64]))
+ aidge_core.register_Tensor([self._name, aidge_core.dtype.int4],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.int4]))
+ aidge_core.register_Tensor([self._name, aidge_core.dtype.uint4],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.uint4]))
+ aidge_core.register_Tensor([self._name, aidge_core.dtype.dual_int4],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.dual_int4]))
+ aidge_core.register_Tensor([self._name, aidge_core.dtype.dual_uint4],
+ aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.dual_uint4]))
@classproperty
def _export_node_registry(cls) -> Dict[str, List['ExportNode']]:
--
GitLab