From 7fef05356ea5fa0449dc3e06b4f043596110fff4 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Fri, 13 Sep 2024 11:48:34 +0000
Subject: [PATCH] Reflect changes due to resaling kernel moved.
---
aidge_export_cpp/export_registry.py | 1 -
aidge_export_cpp/operators.py | 7 +++++++
2 files changed, 7 insertions(+), 1 deletion(-)
diff --git a/aidge_export_cpp/export_registry.py b/aidge_export_cpp/export_registry.py
index 06d9bb5..fe7e80c 100644
--- a/aidge_export_cpp/export_registry.py
+++ b/aidge_export_cpp/export_registry.py
@@ -5,7 +5,6 @@ class ExportLibCpp(ExportLib):
name="export_cpp"
static_files={
str(ROOT / "static" / "Makefile"): "",
- str(ROOT / "static" / "include" / "network" / "rescaling.hpp"): "dnn/include/network",
str(ROOT / "static" / "include" / "network" / "typedefs.hpp"): "dnn/include/network",
str(ROOT / "static" / "include" / "network" / "utils.hpp"): "dnn/include/network",
}
diff --git a/aidge_export_cpp/operators.py b/aidge_export_cpp/operators.py
index b3d6d28..b31fe85 100644
--- a/aidge_export_cpp/operators.py
+++ b/aidge_export_cpp/operators.py
@@ -94,6 +94,7 @@ class ReLUCPP(ExportNodeCpp):
self.include_list = []
self.kernels_to_copy = [
str(ROOT / "kernels" / "activation.hpp"),
+ str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
@@ -118,6 +119,7 @@ class ConvCPP(ExportNodeCpp):
str(ROOT / "kernels" / "convolution.hpp"),
str(ROOT / "kernels" / "macs.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
+ str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
@@ -152,6 +154,7 @@ class PaddedConvCPP(ExportNodeCpp):
str(ROOT / "kernels" / "convolution.hpp"),
str(ROOT / "kernels" / "macs.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
+ str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
@@ -172,6 +175,7 @@ class AddCPP(ExportNodeCpp):
self.kernels_to_copy = [
str(ROOT / "kernels" / "elemwise.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
+ str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
@@ -192,6 +196,7 @@ class SubCPP(ExportNodeCpp):
self.kernels_to_copy = [
str(ROOT / "kernels" / "elemwise.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
+ str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
@@ -213,6 +218,7 @@ class MulCPP(ExportNodeCpp):
self.kernels_to_copy = [
str(ROOT / "kernels" / "elemwise.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
+ str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
@@ -316,6 +322,7 @@ class FcCPP(ExportNodeCpp):
str(ROOT / "kernels" / "fullyconnected.hpp"),
str(ROOT / "kernels" / "macs.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
+ str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
--
GitLab