From 1bbc1dcba657e4f50636ae98d0f4bb82b3865b4a Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Fri, 11 Oct 2024 11:59:31 +0000
Subject: [PATCH] [MINOR] format update.
---
aidge_core/export_utils/export_registry.py | 10 ++--------
aidge_core/export_utils/templates/forward.jinja | 8 ++++----
aidge_core/export_utils/templates/forward_header.jinja | 6 +++---
3 files changed, 9 insertions(+), 15 deletions(-)
diff --git a/aidge_core/export_utils/export_registry.py b/aidge_core/export_utils/export_registry.py
index 6cdba2851..eabc6eb6b 100644
--- a/aidge_core/export_utils/export_registry.py
+++ b/aidge_core/export_utils/export_registry.py
@@ -1,12 +1,6 @@
-from typing import Dict, List, Set, Type
+from typing import Dict, List
import aidge_core
from aidge_core.export_utils import ExportNode
-from abc import ABC
-from enum import Enum
-
-# Language
-LANGUAGE = Enum('LANGUAGE', ['Cpp/C'])
-
class classproperty:
"""Helper class to define class properties,
@@ -35,7 +29,7 @@ class ExportLib(aidge_core.OperatorImpl):
# Value: Path where to copy the file relative to the export root
static_files: Dict[str, str] = {}
# PRIVATE
- # Registry of exportNode, class level dictionary, shared accross all ExportLib
+ # Registry of exportNode, class level dictionary, shared across all ExportLib
_cls_export_node_registry = {}
def __init__(self, operator):
diff --git a/aidge_core/export_utils/templates/forward.jinja b/aidge_core/export_utils/templates/forward.jinja
index a58bd168a..aec5867b1 100644
--- a/aidge_core/export_utils/templates/forward.jinja
+++ b/aidge_core/export_utils/templates/forward.jinja
@@ -19,12 +19,12 @@ static {{mem_ctype}} mem[{{peak_mem}}];
{# Forward function #}
{#- Support multiple inputs with different datatypes and multiple outputs with different datatypes -#}
void {{ func_name }} (
- {%- for i in range(inputs_name | length) %}
+ {%- for i in range(inputs_name | length) -%}
const {{ inputs_dtype[i] }}* {{ inputs_name[i] }},
- {% endfor -%}
- {%- for o in range(outputs_name | length) %}
+ {%- endfor -%}
+ {%- for o in range(outputs_name | length) -%}
{{ outputs_dtype[o] }}** {{ outputs_name[o] }}_ptr{% if not loop.last %}, {% endif %}
- {% endfor -%})
+ {%- endfor -%})
{
{%- for action in actions %}
diff --git a/aidge_core/export_utils/templates/forward_header.jinja b/aidge_core/export_utils/templates/forward_header.jinja
index 8530d91e9..574f53238 100644
--- a/aidge_core/export_utils/templates/forward_header.jinja
+++ b/aidge_core/export_utils/templates/forward_header.jinja
@@ -11,12 +11,12 @@ extern "C" {
{%- endfor %}
void {{ func_name }} (
- {%- for i in range(inputs_name | length) %}
+ {%- for i in range(inputs_name | length) -%}
const {{ inputs_dtype[i] }}* {{ inputs_name[i] }},
- {% endfor -%}
+ {%- endfor -%}
{%- for o in range(outputs_name | length) %}
{{ outputs_dtype[o] }}** {{ outputs_name[o] }}{% if not loop.last %}, {% endif %}
- {% endfor -%});
+ {%- endfor -%});
#ifdef __cplusplus
}
--
GitLab