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