1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
package = "nnx"
version = "1.0-1"
source = {
url = "nnx-1.0-1.tgz"
}
description = {
summary = "An extension to Torch7's nn package.",
detailed = [[
This package provides extra trainable modules,
which naturally extend the nn package.
Some of those might get marged into the original
nn package, at some point. For this reason,
all the modules from nnx are appended to nn.
]],
homepage = "",
license = "MIT/X11" -- or whatever you like
}
dependencies = {
"lua >= 5.1",
"torch",
"sys",
"xlua"
}
build = {
type = "cmake",
cmake = [[
cmake_minimum_required(VERSION 2.8)
set (CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR})
# infer path for Torch7
string (REGEX REPLACE "(.*)lib/luarocks/rocks.*" "\\1" TORCH_PREFIX "${CMAKE_INSTALL_PREFIX}" )
message (STATUS "Found Torch7, installed in: " ${TORCH_PREFIX})
find_package (Torch REQUIRED)
find_package (OpenMP REQUIRED)
if (OPENMP_FOUND)
message (STATUS "OpenMP Found with compiler flag : ${OpenMP_C_FLAGS}")
set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
endif (OPENMP_FOUND)
set (CMAKE_INSTALL_RPATH_USE_LINK_PATH TRUE)
include_directories (${TORCH_INCLUDE_DIR} ${PROJECT_SOURCE_DIR})
add_library (nnx SHARED init.c)
link_directories (${TORCH_LIBRARY_DIR})
target_link_libraries (nnx ${TORCH_LIBRARIES})
install_files(/lua/nnx init.lua)
install_files(/lua/nnx Abs.lua)
install_files(/lua/nnx ConfusionMatrix.lua)
install_files(/lua/nnx HardShrink.lua)
install_files(/lua/nnx Narrow.lua)
install_files(/lua/nnx Power.lua)
install_files(/lua/nnx Square.lua)
install_files(/lua/nnx Sqrt.lua)
install_files(/lua/nnx Threshold.lua)
install_files(/lua/nnx OmpModule.lua)
install_files(/lua/nnx SpatialConvolutionTable.lua)
install_files(/lua/nnx SpatialLogSoftMax.lua)
install_files(/lua/nnx SpatialMaxPooling.lua)
install_files(/lua/nnx SpatialLinear.lua)
install_files(/lua/nnx SpatialPadding.lua)
install_files(/lua/nnx SpatialNormalization.lua)
install_files(/lua/nnx SpatialUpSampling.lua)
install_files(/lua/nnx SpatialReSampling.lua)
install_files(/lua/nnx SuperCriterion.lua)
install_files(/lua/nnx SpatialCriterion.lua)
install_files(/lua/nnx Trainer.lua)
install_files(/lua/nnx StochasticTrainer.lua)
install_files(/lua/nnx DataSet.lua)
install_files(/lua/nnx DataList.lua)
install_files(/lua/nnx CMulTable.lua)
install_files(/lua/nnx CAddTable.lua)
install_files(/lua/nnx CDivTable.lua)
install_files(/lua/nnx CSubTable.lua)
install_files(/lua/nnx Replicate.lua)
install_files(/lua/nnx SpatialFovea.lua)
install_files(/lua/nnx SpatialMSECriterion.lua)
install_files(/lua/nnx SpatialClassNLLCriterion.lua)
install_files(/lua/nnx SparseCriterion.lua)
install_files(/lua/nnx SpatialSparseCriterion.lua)
install_files(/lua/nnx SpatialGraph.lua)
add_subdirectory (test)
install_targets(/lib nnx)
]],
variables = {
CMAKE_INSTALL_PREFIX = "$(PREFIX)"
}
}
|
