Skip to content
Snippets Groups Projects
Commit bc36561a authored by Maxence Naud's avatar Maxence Naud
Browse files

Path to Operator header file should be specified by hand because the file name...

Path to Operator header file should be specified by hand because the file name may not be exactly named like the associated Operator's type
parent e3207217
No related branches found
No related tags found
3 merge requests!279v0.4.0,!253v0.4.0,!163Export refactor
...@@ -93,7 +93,7 @@ def serialize_to_cpp(export_folder: str, ...@@ -93,7 +93,7 @@ def serialize_to_cpp(export_folder: str,
node, None) node, None)
set_operator.add(node.type()) # set_operator.add(node.type())
# TODO: list_configs and list_actions don't need to be passed by argument # TODO: list_configs and list_actions don't need to be passed by argument
# Export the configuration # Export the configuration
......
...@@ -11,7 +11,7 @@ class Conv(ExportNodeCpp): ...@@ -11,7 +11,7 @@ class Conv(ExportNodeCpp):
ROOT_EXPORT / "templates/attributes/conv.jinja") ROOT_EXPORT / "templates/attributes/conv.jinja")
self.forward_template = str( self.forward_template = str(
ROOT_EXPORT /"templates/graph_ctor/conv.jinja") ROOT_EXPORT /"templates/graph_ctor/conv.jinja")
self.include_list = [] self.include_list = ["aidge/operator/Conv.hpp"]
self.kernels_to_copy = [] self.kernels_to_copy = []
self.config_path = "include/attributes" self.config_path = "include/attributes"
self.config_extension = "hpp" self.config_extension = "hpp"
...@@ -12,7 +12,7 @@ class FC(ExportNodeCpp): ...@@ -12,7 +12,7 @@ class FC(ExportNodeCpp):
ROOT_EXPORT / "templates/attributes/fc.jinja") ROOT_EXPORT / "templates/attributes/fc.jinja")
self.forward_template = str( self.forward_template = str(
ROOT_EXPORT / "templates/graph_ctor/fc.jinja") ROOT_EXPORT / "templates/graph_ctor/fc.jinja")
self.include_list = [] self.include_list = ["aidge/operator/FC.hpp"]
self.kernels_to_copy = [] self.kernels_to_copy = []
self.config_path = "include/attributes" self.config_path = "include/attributes"
self.config_extension = "hpp" self.config_extension = "hpp"
...@@ -11,7 +11,7 @@ class MaxPooling(ExportNodeCpp): ...@@ -11,7 +11,7 @@ class MaxPooling(ExportNodeCpp):
ROOT_EXPORT / "templates/attributes/maxpooling.jinja") ROOT_EXPORT / "templates/attributes/maxpooling.jinja")
self.forward_template = str( self.forward_template = str(
ROOT_EXPORT / "templates/graph_ctor/maxpooling.jinja") ROOT_EXPORT / "templates/graph_ctor/maxpooling.jinja")
self.include_list = [] self.include_list = ["aidge/operator/MaxPooling.hpp"]
self.kernels_to_copy = [] self.kernels_to_copy = []
self.config_path = "include/attributes" self.config_path = "include/attributes"
self.config_extension = "hpp" self.config_extension = "hpp"
...@@ -23,7 +23,7 @@ class Producer(ExportNodeCpp): ...@@ -23,7 +23,7 @@ class Producer(ExportNodeCpp):
ROOT_EXPORT / "templates/graph_ctor/producer.jinja") ROOT_EXPORT / "templates/graph_ctor/producer.jinja")
self.attributes["tensor_name"] = f"{child.name()}_{in_idx}" self.attributes["tensor_name"] = f"{child.name()}_{in_idx}"
self.attributes["values"] = str(self.operator.get_output(0)) self.attributes["values"] = str(self.operator.get_output(0))
self.include_list = [] self.include_list = ["aidge/operator/Producer.hpp"]
self.kernels_to_copy = [] self.kernels_to_copy = []
self.config_path = "include/attributes" self.config_path = "include/attributes"
self.config_extension = "hpp" self.config_extension = "hpp"
......
...@@ -10,5 +10,5 @@ class ReLU(ExportNodeCpp): ...@@ -10,5 +10,5 @@ class ReLU(ExportNodeCpp):
self.config_template = "" self.config_template = ""
self.forward_template = str( self.forward_template = str(
ROOT_EXPORT / "templates/graph_ctor/relu.jinja") ROOT_EXPORT / "templates/graph_ctor/relu.jinja")
self.include_list = [] self.include_list = ["aidge/operator/ReLU.hpp"]
self.kernels_to_copy = [] self.kernels_to_copy = []
...@@ -10,5 +10,5 @@ class Sub(ExportNodeCpp): ...@@ -10,5 +10,5 @@ class Sub(ExportNodeCpp):
self.config_template = "" self.config_template = ""
self.forward_template = str( self.forward_template = str(
ROOT_EXPORT / "templates/graph_ctor/sub.jinja") ROOT_EXPORT / "templates/graph_ctor/sub.jinja")
self.include_list = [] self.include_list = ["aidge/operator/Sub.hpp"]
self.kernels_to_copy = [] self.kernels_to_copy = []
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