blob: cebddfbfc4a0da38140cd34d5788d74894b0dce3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
CMAKE_MINIMUM_REQUIRED(VERSION 2.6 FATAL_ERROR)
CMAKE_POLICY(VERSION 2.6)
INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR}/../torch7/lib/TH)
INCLUDE_DIRECTORIES(${CMAKE_CURRENT_BINARY_DIR}/../torch7/lib/TH)
ADD_SUBDIRECTORY(lib)
FILE(STRINGS lib/THNN/generic/THNN.h THNN_headers NEWLINE_CONSUME)
FILE(WRITE THNN_h.lua "return [[")
FILE(APPEND THNN_h.lua ${THNN_headers})
FILE(APPEND THNN_h.lua "]]")
FILE(GLOB luasrc *.lua)
ADD_TORCH_PACKAGE(nn "" "${luasrc}")
|
