Skip to content
Snippets Groups Projects
CMakeLists.txt 1.22 KiB
Newer Older
  • Learn to ignore specific revisions
  • 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