Skip to content
Snippets Groups Projects
Commit 7816487a authored by Axel Farrugia's avatar Axel Farrugia
Browse files

[Feat](Exports) Add list nodes names to forward generation

parent 76b61da0
No related branches found
No related tags found
No related merge requests found
Pipeline #59721 canceled
......@@ -157,6 +157,11 @@ def scheduler_export(scheduler, export_folder_path: str, export_lib: ExportLib =
if export_lib.forward_template != None:
forward_template = export_lib.forward_template
list_node_names = []
for node in list_forward_nodes:
if node.type != "Producer":
list_node_names.append(node.name())
generate_file(
str(dnn_folder / "src" / "forward.cpp"),
forward_template,
......@@ -173,7 +178,8 @@ def scheduler_export(scheduler, export_folder_path: str, export_lib: ExportLib =
inputs_dtype=inputs_dtype,
outputs_name=outputs_name,
outputs_dtype=outputs_dtype,
test_mode=test_mode
test_mode=test_mode,
list_node_names=list_node_names
)
forward_header_template = str(ROOT / "templates" / "forward_header.jinja")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment