From 9145801371105a5c8b8df9c095ee3feb88bf8089 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Tue, 23 Jul 2024 11:41:19 +0000
Subject: [PATCH] Main.cpp now include header with input name.
---
aidge_core/export_utils/templates/forward_header.jinja | 2 +-
aidge_core/export_utils/templates/main.jinja | 6 +++++-
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/aidge_core/export_utils/templates/forward_header.jinja b/aidge_core/export_utils/templates/forward_header.jinja
index 1e14cdea6..86333e6c9 100644
--- a/aidge_core/export_utils/templates/forward_header.jinja
+++ b/aidge_core/export_utils/templates/forward_header.jinja
@@ -7,7 +7,7 @@
{%- endfor %}
{% for func in functions %}
-{{ func }}
+{{ func }};
{% endfor %}
#endif /* DNN_HPP */
diff --git a/aidge_core/export_utils/templates/main.jinja b/aidge_core/export_utils/templates/main.jinja
index 06945222f..40ca12e0a 100644
--- a/aidge_core/export_utils/templates/main.jinja
+++ b/aidge_core/export_utils/templates/main.jinja
@@ -1,7 +1,11 @@
#include <iostream>
#include "forward.hpp"
-#include "inputs.h"
+{% for name in inputs_name %}
+#include "{{ name }}.h"
+{% endfor %}
+
+
int main()
{
--
GitLab