From bc96682344b779cabe6cfd97fddd7ec98af0bb78 Mon Sep 17 00:00:00 2001
From: Axel Farrugia <axel.farrugia@cea.fr>
Date: Tue, 22 Oct 2024 10:16:32 +0200
Subject: [PATCH] [Feat](Exports) Add labels handling in main generation
---
aidge_core/export_utils/scheduler_export.py | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/aidge_core/export_utils/scheduler_export.py b/aidge_core/export_utils/scheduler_export.py
index b8a0e1f5d..df0b4a385 100644
--- a/aidge_core/export_utils/scheduler_export.py
+++ b/aidge_core/export_utils/scheduler_export.py
@@ -6,7 +6,7 @@ from aidge_core.export_utils import ExportLib, generate_file, copy_file
from typing import List, Tuple
-def scheduler_export(scheduler, export_folder_path: str, export_lib: ExportLib = None, memory_manager=None, memory_manager_args=None) -> None:
+def scheduler_export(scheduler, export_folder_path: str, export_lib: ExportLib = None, memory_manager=None, memory_manager_args=None, labels=False) -> None:
graphview = scheduler.graph_view()
export_folder = Path().absolute() / export_folder_path
@@ -137,11 +137,10 @@ def scheduler_export(scheduler, export_folder_path: str, export_lib: ExportLib =
if len(outputs_name) != len(outputs_dtype) or len(outputs_name) != len(outputs_size):
raise RuntimeError("FATAL: Output args list does not have the same length this is an internal bug.")
- if export_lib is not None:
- if export_lib.main_jinja_path is not None:
- main_jinja_path = export_lib.main_jinja_path
- else :
- main_jinja_path = str(ROOT / "templates" / "main.jinja")
+ if export_lib is not None and export_lib.main_jinja_path is not None:
+ main_jinja_path = export_lib.main_jinja_path
+ else :
+ main_jinja_path = str(ROOT / "templates" / "main.jinja")
generate_file(
str(export_folder / "main.cpp"),
@@ -150,7 +149,8 @@ def scheduler_export(scheduler, export_folder_path: str, export_lib: ExportLib =
inputs_name=inputs_name,
outputs_name=outputs_name,
outputs_dtype=outputs_dtype,
- outputs_size=outputs_size
+ outputs_size=outputs_size,
+ labels=labels
)
if export_lib is not None:
--
GitLab