diff --git a/aidge_export_arm_cortexm/operators.py b/aidge_export_arm_cortexm/operators.py index bbc111d3d12845e8c20f3fc367d943b7fafa5eb5..24611dde638109bcd19790870cf73c561e46601f 100644 --- a/aidge_export_arm_cortexm/operators.py +++ b/aidge_export_arm_cortexm/operators.py @@ -188,7 +188,7 @@ class ReLU_ARMCortexM(ExportNodeCpp): self.forward_template = str(ROOT / "_Aidge_Arm" / "templates" / "forward_call" / "relu.jinja") self.include_list = [] self.kernels_to_copy = [ - str(ROOT / "_Aidge_Arm" / "kernels" / "Relu" / "aidge_relu_float32.c"), + str(ROOT / "_Aidge_Arm" / "kernels" / "Relu" / "aidge_relu_float32.h"), ] @classmethod def exportable(cls, node): @@ -207,7 +207,9 @@ class Conv_ARMCortexM(ExportNodeCpp): self.config_template = str(ROOT / "_Aidge_Arm" / "templates" / "configuration" / "conv_config.jinja") self.forward_template = str(ROOT / "_Aidge_Arm" / "templates" / "forward_call" / "conv_kernel.jinja") self.include_list = [] - self.kernels_to_copy = [] + self.kernels_to_copy = [ + str(ROOT / "_Aidge_Arm" / "kernels" / "Convolution" / "Conv.hpp") + ] @classmethod def exportable(cls, node): return True # TODO add check i/o NCHW @@ -233,7 +235,9 @@ class PaddedConv_ARMCortexM(ExportNodeCpp): self.config_template = str(ROOT / "_Aidge_Arm" / "templates" / "configuration" / "conv_config.jinja") self.forward_template = str(ROOT / "_Aidge_Arm" / "templates" / "forward_call" / "conv_kernel.jinja") self.include_list = [] - self.kernels_to_copy = [] + self.kernels_to_copy = [ + str(ROOT / "_Aidge_Arm" / "kernels" / "Convolution" / "Conv.hpp") + ] @classmethod def exportable(cls, node): return True # TODO add check i/o NCHW @@ -253,7 +257,9 @@ class Pooling_ARMCortexM(ExportNodeCpp): self.config_template = str(ROOT / "_Aidge_Arm" / "templates" / "configuration" / "pool_config.jinja") self.forward_template = str(ROOT / "_Aidge_Arm" / "templates" / "forward_call" / "pool_kernel.jinja") self.include_list = [] - self.kernels_to_copy = [] + self.kernels_to_copy = [ + str(ROOT / "_Aidge_Arm" / "kernels" / "Pooling" / "Pooling.hpp") + ] self.kernel = node.get_operator().attr.kernel_dims self.stride = node.get_operator().attr.stride_dims @classmethod @@ -273,7 +279,9 @@ class FC_ARMCortexM(ExportNodeCpp): self.config_template = str(ROOT / "_Aidge_Arm" / "templates" / "configuration" / "fc_config.jinja") self.forward_template = str(ROOT / "_Aidge_Arm" / "templates" / "forward_call" / "fc_kernel.jinja") self.include_list = [] - self.kernels_to_copy = [] + self.kernels_to_copy = [ + str(ROOT / "_Aidge_Arm" / "kernels" / "FullyConnected" / "Fc.hpp") + ] @classmethod def exportable(cls, node): return True # TODO add check i/o NCHW