diff options
| author | Determinant <ted.sybil@gmail.com> | 2015-06-02 20:28:16 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2015-06-02 20:28:16 +0800 |
| commit | 74d9e9e7371c80394698fb9805cbf0cbde67a8f3 (patch) | |
| tree | 36b070f1fcfa2be8fc80c50b7a221862a0dfd14a /Makefile | |
| parent | 60083f2e51935ce55cec7a4c39d1724a16d9c769 (diff) | |
add ParamRepo, LayerRepo, DAGLayer
Diffstat (limited to 'Makefile')
| -rw-r--r-- | Makefile | 5 |
1 files changed, 3 insertions, 2 deletions
@@ -8,7 +8,8 @@ LIBS := libnerv.so LUA_LIBS := matrix/init.lua io/init.lua nerv.lua \ pl/utils.lua pl/compat.lua \ layer/init.lua layer/affine.lua layer/sigmoid.lua layer/softmax_ce.lua \ - layer/window.lua layer/bias.lua + layer/window.lua layer/bias.lua \ + nn/init.lua nn/layer_repo.lua nn/param_repo.lua nn/layer_dag.lua INCLUDE := -I build/luajit-2.0/include/luajit-2.0/ -DLUA_USE_APICHECK CUDA_BASE := /usr/local/cuda-6.5 CUDA_INCLUDE := -I $(CUDA_BASE)/include/ @@ -18,7 +19,7 @@ CFLAGS := -Wall -Wextra OBJ_DIR := $(BUILD_DIR)/objs LUA_DIR := $(BUILD_DIR)/lua LIB_DIR := $(BUILD_DIR)/lib -SUBDIR := matrix io layer examples pl +SUBDIR := matrix io layer examples pl nn NVCC := $(CUDA_BASE)/bin/nvcc NVCC_FLAGS := -Xcompiler -fPIC,-Wall,-Wextra |