From a848e35883cd21acc1d072dd814fd82427f19309 Mon Sep 17 00:00:00 2001 From: Mickael GUIBERT <mickael.guibert@cea.fr> Date: Tue, 1 Apr 2025 08:07:26 +0000 Subject: [PATCH] [Feat] Add jinja file for forward build --- .../configuration/rescaling_config.jinja | 15 +++++++++++++++ .../kernel_forward/rescaling_forward.jinja | 9 +++++++++ 2 files changed, 24 insertions(+) create mode 100644 aidge_export_cpp/templates/configuration/rescaling_config.jinja create mode 100644 aidge_export_cpp/templates/kernel_forward/rescaling_forward.jinja diff --git a/aidge_export_cpp/templates/configuration/rescaling_config.jinja b/aidge_export_cpp/templates/configuration/rescaling_config.jinja new file mode 100644 index 0000000..ee9c69e --- /dev/null +++ b/aidge_export_cpp/templates/configuration/rescaling_config.jinja @@ -0,0 +1,15 @@ +{#- For name header -#} +#ifndef {{ name|upper }}_LAYER_H +#define {{ name|upper }}_LAYER_H + +{# For layer configuration -#} +{% include "./_def_io.jinja" %} +{% include "./_meminfo.jinja" %} + +#define {{ name|upper }}_NB_DATA {{ in_chan[0] * in_height[0] * in_width[0] }} + +// Activation +#define {{ name|upper }}_ACTIVATION {{ activation }} +{% include "./_rescaling.jinja" %} + +#endif /* {{ name|upper }}_LAYER_H */ diff --git a/aidge_export_cpp/templates/kernel_forward/rescaling_forward.jinja b/aidge_export_cpp/templates/kernel_forward/rescaling_forward.jinja new file mode 100644 index 0000000..ce4ffb8 --- /dev/null +++ b/aidge_export_cpp/templates/kernel_forward/rescaling_forward.jinja @@ -0,0 +1,9 @@ +{% filter indent(width=4, first=False) %} +{% include "./_mem_offset.jinja" %} +rescaling_forward<{{name|upper}}_NB_DATA, + {{name|upper}}_ACTIVATION> + ({{in_name[0]}}, + {{out_name[0]}}, + {{name|upper}}_RESCALING); +{% include "./_save_outputs.jinja" %} +{% endfilter %} -- GitLab