Skip to content
Snippets Groups Projects
CMakeLists.txt 1.22 KiB
Newer Older
Franck Dary's avatar
Franck Dary committed
cmake_minimum_required(VERSION 2.8.7)
project(test_torch)

add_compile_definitions(BOOST_DISABLE_THREADS)

Franck Dary's avatar
Franck Dary committed
find_package(Torch REQUIRED)
find_package(Boost 1.53.0 REQUIRED COMPONENTS program_options)
Franck Dary's avatar
Franck Dary committed

include_directories(SYSTEM ${TORCH_INCLUDE_DIRS})

add_library(Torch SHARED IMPORTED)
set_target_properties(Torch PROPERTIES IMPORTED_LOCATION ${TORCH_LIBRARIES})
add_library(Boost SHARED IMPORTED)
set_target_properties(Boost PROPERTIES IMPORTED_LOCATION ${Boost_PROGRAM_OPTIONS_LIBRARY_RELEASE})

string(APPEND CMAKE_INSTALL_RPATH ":${TORCH_INSTALL_PREFIX}/lib")
Franck Dary's avatar
Franck Dary committed

set(CMAKE_VERBOSE_MAKEFILE 0)
set(CMAKE_CXX_STANDARD 17)
Franck Dary's avatar
Franck Dary committed

if(NOT CMAKE_BUILD_TYPE)
Franck Dary's avatar
Franck Dary committed
#  set(CMAKE_BUILD_TYPE Debug)
  set(CMAKE_BUILD_TYPE Release)
Franck Dary's avatar
Franck Dary committed
endif()

Franck Dary's avatar
Franck Dary committed
set(CMAKE_CXX_FLAGS_DEBUG "-g3")
Franck Dary's avatar
Franck Dary committed
set(CMAKE_CXX_FLAGS_RELEASE "-Ofast")

include_directories(fmt/include)
Franck Dary's avatar
Franck Dary committed
include_directories(common/include)
include_directories(reading_machine/include)
include_directories(torch_modules/include)
Franck Dary's avatar
Franck Dary committed
include_directories(trainer/include)
Franck Dary's avatar
Franck Dary committed
include_directories(decoder/include)
Franck Dary's avatar
Franck Dary committed
include_directories(utf8)

add_subdirectory(fmt)
Franck Dary's avatar
Franck Dary committed
add_subdirectory(common)
add_subdirectory(reading_machine)
add_subdirectory(torch_modules)
Franck Dary's avatar
Franck Dary committed
add_subdirectory(trainer)
Franck Dary's avatar
Franck Dary committed
add_subdirectory(decoder)
Franck Dary's avatar
Franck Dary committed