From 8e616e065501be8b402b310450c8652226b7b5d9 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Mon, 6 Jan 2020 14:12:08 +0100
Subject: [PATCH] Added directory torch_modules

---
 CMakeLists.txt                         | 2 ++
 reading_machine/include/Classifier.hpp | 1 +
 torch_modules/CMakeLists.txt           | 4 ++++
 torch_modules/include/MLP.hpp          | 9 +++++++++
 torch_modules/src/MLP.cpp              | 1 +
 5 files changed, 17 insertions(+)
 create mode 100644 torch_modules/CMakeLists.txt
 create mode 100644 torch_modules/include/MLP.hpp
 create mode 100644 torch_modules/src/MLP.cpp

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 9bd48f1..54f5cb6 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -25,10 +25,12 @@ set(CMAKE_CXX_FLAGS_RELEASE "-Ofast")
 include_directories(fmt/include)
 include_directories(common/include)
 include_directories(reading_machine/include)
+include_directories(torch_modules/include)
 include_directories(utf8)
 
 add_subdirectory(fmt)
 add_subdirectory(common)
 add_subdirectory(dev)
 add_subdirectory(reading_machine)
+add_subdirectory(torch_modules)
 
diff --git a/reading_machine/include/Classifier.hpp b/reading_machine/include/Classifier.hpp
index 24cc9aa..41cac65 100644
--- a/reading_machine/include/Classifier.hpp
+++ b/reading_machine/include/Classifier.hpp
@@ -2,6 +2,7 @@
 #define CLASSIFIER__H
 
 #include <string>
+#include <torch/torch.h>
 #include "TransitionSet.hpp"
 
 class Classifier
diff --git a/torch_modules/CMakeLists.txt b/torch_modules/CMakeLists.txt
new file mode 100644
index 0000000..6ccb7ce
--- /dev/null
+++ b/torch_modules/CMakeLists.txt
@@ -0,0 +1,4 @@
+FILE(GLOB SOURCES src/*.cpp)
+
+add_library(torch_modules STATIC ${SOURCES})
+target_link_libraries(torch_modules "${TORCH_LIBRARIES}")
diff --git a/torch_modules/include/MLP.hpp b/torch_modules/include/MLP.hpp
new file mode 100644
index 0000000..ddbd44f
--- /dev/null
+++ b/torch_modules/include/MLP.hpp
@@ -0,0 +1,9 @@
+#ifndef MLP__H
+#define MLP__H
+
+class MLPImpl
+{
+
+};
+
+#endif
diff --git a/torch_modules/src/MLP.cpp b/torch_modules/src/MLP.cpp
new file mode 100644
index 0000000..0a5a320
--- /dev/null
+++ b/torch_modules/src/MLP.cpp
@@ -0,0 +1 @@
+#include "MLP.hpp"
-- 
GitLab