Fix CPP export for Add Sub and add export for Mul

Context

Add, Mul and Sub did not export correctly through aidge_export_cpp.

Fixed and add export for Mul

Modified files

Added in aidge_expot_cpp/operators.py;

• fixes in AddCPP and SubCPP class
• implementation of MulCPP class + registration of operator Mul

Detailed major modifications

No major modification, except the addition of the support for Mul export.

TODO

No todo forecasted.

Note though that the code for the element wise operators could be factorized out in the future.

Note also, that there is for now unit tests missing for operator exports, could contribute to this in the future.

aidge_export_cpp/operators.py

@@ -238,6 +238,8 @@ class AddCPP(ExportNode):
 
         copyfile(str(ROOT / "kernels" / "elemwise.hpp"),
                  str(export_folder / "include" / "kernels"))
+        copyfile(str(ROOT / "kernels" / "activation.hpp"),
+                 str(export_folder / "include" / "kernels"))
 
         generate_file(
             str(export_folder / "layers" / f"{self.name}.h"),
@@ -251,13 +253,13 @@ class AddCPP(ExportNode):
         return list_configs
 
     def forward(self, list_actions:list):
-
-        list_actions.append(set_up_output(self.name, "float"))
+        if not self.is_last:
+            list_actions.append(set_up_output(self.name, "float"))
         list_actions.append(generate_str(
             str(ROOT / "templates" / "kernel_forward" / "elemwise_forward.jinja"),
             name=self.name,
-            inputs1_name=self.parents[0].name() if self.parents[0] else self.name + "_input1",
-            inputs2_name=self.parents[1].name() if self.parents[1] else self.name + "_input2",
+            inputs1_name=self.inputs[0].name() if self.inputs[0] else self.name + "_input1",
+            inputs2_name=self.inputs[1].name() if self.inputs[1] else self.name + "_input2",
             output_name=self.name
         ))
         return list_actions
@@ -272,6 +274,9 @@ class SubCPP(ExportNode):
         list_configs.append("kernels/elemwise.hpp")
         copyfile(str(ROOT / "kernels" / "elemwise.hpp"),
                  str(export_folder / "include" / "kernels"))
+        copyfile(str(ROOT / "kernels" / "activation.hpp"),
+                 str(export_folder / "include" / "kernels"))
+
         generate_file(
             str(export_folder / "layers" / f"{self.name}.h"),
             str(ROOT / "templates" / "configuration" / "elemwise_config.jinja"),
@@ -284,8 +289,46 @@ class SubCPP(ExportNode):
         return list_configs
 
     def forward(self, list_actions:list):
+        if not self.is_last:
+            list_actions.append(set_up_output(self.name, "float"))
+
+        list_actions.append(generate_str(
+            str(ROOT / "templates" / "kernel_forward" / "elemwise_forward.jinja"),
+            name=self.name,
+            inputs1_name=self.inputs[0].name() if self.inputs[0] else self.name + "_input1",
+            inputs2_name=self.inputs[1].name() if self.inputs[1] else self.name + "_input2",
+            output_name=self.name
+        ))
+        return list_actions
+
+@operator_register("Mul")
+class MulCPP(ExportNode):
+    def __init__(self, node):
+        super().__init__(node)
+
+    def export(self, export_folder:str, list_configs:list):
+        list_configs.append(f"layers/{self.name}.h")
+        list_configs.append("kernels/elemwise.hpp")
+        copyfile(str(ROOT / "kernels" / "elemwise.hpp"),
+                 str(export_folder / "include" / "kernels"))
+        copyfile(str(ROOT / "kernels" / "activation.hpp"),
+                 str(export_folder / "include" / "kernels"))
+
+        generate_file(
+            str(export_folder / "layers" / f"{self.name}.h"),
+            str(ROOT / "templates" / "configuration" / "elemwise_config.jinja"),
+            name=self.name,
+            nb_elts=np.prod(self.inputs_dims[0]),
+            activation="Linear",
+            elemwise_op="Mul",
+            rescaling="NoScaling")
+
+        return list_configs
+
+    def forward(self, list_actions:list):
+        if not self.is_last:
+            list_actions.append(set_up_output(self.name, "float"))
 
-        list_actions.append(set_up_output(self.name, "float"))
         list_actions.append(generate_str(
             str(ROOT / "templates" / "kernel_forward" / "elemwise_forward.jinja"),
             name=self.name,