From 56be9489c374899e055f46792154d1ea345ec2c1 Mon Sep 17 00:00:00 2001
From: Axel Farrugia <axel.farrugia@cea.fr>
Date: Thu, 24 Apr 2025 10:46:10 +0200
Subject: [PATCH] [Refactor] node.ignore is now a flag instead of a required
attribute
---
aidge_export_cpp/export_utils.py | 1 -
aidge_export_cpp/operators/Producer.py | 10 +---------
2 files changed, 1 insertion(+), 10 deletions(-)
diff --git a/aidge_export_cpp/export_utils.py b/aidge_export_cpp/export_utils.py
index a7f32bb..9ce15f5 100644
--- a/aidge_export_cpp/export_utils.py
+++ b/aidge_export_cpp/export_utils.py
@@ -167,7 +167,6 @@ def exclude_unwanted_producers(model):
nodes_to_ignore = ["Mul", "BitShift"]
for node in model.get_nodes():
- node.attributes().ignore = False
if node.type() == "Producer":
children_nodes = [n.type() for n in node.get_children()]
for node_type in nodes_to_ignore:
diff --git a/aidge_export_cpp/operators/Producer.py b/aidge_export_cpp/operators/Producer.py
index 6c75471..627dcb2 100644
--- a/aidge_export_cpp/operators/Producer.py
+++ b/aidge_export_cpp/operators/Producer.py
@@ -47,15 +47,7 @@ class ProducerCPP(ExportNode):
def __init__(self, node, mem_info):
super().__init__(node, mem_info)
self.values = np.array(self.operator.get_output(0))
- if node.attributes().has_attr("ignore"):
- self.ignore = node.attributes().ignore
- else:
- self.ignore = False
-
- if node.attributes().has_attr("ignore"):
- self.ignore = node.attributes().ignore
- else:
- self.ignore = False
+ self.ignore = node.attributes().has_attr("ignore")
if len(self.values.shape) == 4: # Note: export in HWC
self.values = np.transpose(self.values, (0, 2, 3, 1))
--
GitLab