From ddc91ac918f5d98b3308233e0b21043409b778f7 Mon Sep 17 00:00:00 2001
From: Vincent TEMPLIER <vincent.templier@cea.fr>
Date: Thu, 27 Jul 2023 08:09:33 +0000
Subject: [PATCH] Add aidge_cpu header file

---
 aidge/_CPU/include/aidge_cpu.hpp         | 6 ++++++
 aidge/_CPU/python_binding/pybind_cpu.cpp | 3 +--
 2 files changed, 7 insertions(+), 2 deletions(-)
 create mode 100644 aidge/_CPU/include/aidge_cpu.hpp

diff --git a/aidge/_CPU/include/aidge_cpu.hpp b/aidge/_CPU/include/aidge_cpu.hpp
new file mode 100644
index 00000000..4b0b1ca5
--- /dev/null
+++ b/aidge/_CPU/include/aidge_cpu.hpp
@@ -0,0 +1,6 @@
+#include "data/TensorImpl.hpp"
+#include "operator/AddImpl.hpp"
+#include "operator/ConvImpl.hpp"
+#include "operator/FCImpl.hpp"
+#include "operator/ProducerImpl.hpp"
+#include "operator/ReLUImpl.hpp"
\ No newline at end of file
diff --git a/aidge/_CPU/python_binding/pybind_cpu.cpp b/aidge/_CPU/python_binding/pybind_cpu.cpp
index 6b5b7cd1..f8267163 100644
--- a/aidge/_CPU/python_binding/pybind_cpu.cpp
+++ b/aidge/_CPU/python_binding/pybind_cpu.cpp
@@ -1,7 +1,6 @@
 #include <pybind11/pybind11.h>
 // Need to call this header to register every impl
-#include "data/TensorImpl.hpp"
-#include "operator/AddImpl.hpp"
+#include "aidge_cpu.hpp"
 
 namespace py = pybind11;
 
-- 
GitLab