blob: 1f1ba1e726674114dc9d46019a6840c8d3425505 (
plain) (
blame)
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
|
.PHONY: build install clean
SHELL := /bin/bash
BUILD_DIR := $(CURDIR)/build
LUA_DIR = $(INST_LUADIR)/fastnn
CUDA_BASE := /usr/local/cuda-6.5
INC_PATH := -I $(LUA_BINDIR)/../include/nerv
CUDA_INCLUDE := -I $(CUDA_BASE)/include/
LIB_PATH := $(LUA_BINDIR)/../lib
OBJ_DIR := $(BUILD_DIR)/objs
SUBDIR := lib io threads device
OBJS := init.o lib/modelsync.o lib/ModelSync.o io/example.o io/Example.o device/device.o device/Device.o
LIBS := $(INST_LIBDIR)/libfastnn.so
LUA_LIBS := init.lua lib/modelsync.lua io/example.lua device/device.lua
INCLUDE := -I $(LUA_INCDIR) $(INC_PATH) $(CUDA_INCLUDE) -DLUA_USE_APICHECK -DUSE_PTHREAD_THREADS
LDFLAGS := -L$(CUDA_BASE)/lib64/ -Wl,-rpath=$(CUDA_BASE)/lib64/ -lcudart -lcublas
OBJS := $(addprefix $(OBJ_DIR)/,$(OBJS))
OBJ_SUBDIR := $(addprefix $(OBJ_DIR)/,$(SUBDIR))
LUA_SUBDIR := $(addprefix $(LUA_DIR)/,$(SUBDIR))
LUA_LIBS := $(addprefix $(LUA_DIR)/,$(LUA_LIBS))
CFLAGS := -Wall -Wextra -g -O0
build: $(OBJ_DIR) $(OBJ_SUBDIR) $(OBJS)
install: $(LIBS) $(LUA_DIR) $(LUA_SUBDIR) $(LUA_LIBS)
#$(SUBMODULE): ECHO
# cd threads; $(LUA_BINDIR)/luarocks make
$(OBJ_DIR) $(LUA_DIR) $(OBJ_SUBDIR) $(LUA_SUBDIR):
-mkdir -p $@
$(LUA_DIR)/%.lua: %.lua
cp $< $@
$(OBJ_DIR)/io/Example.o: io/Example.cpp
gcc -c -o $@ $< $(INCLUDE) -fPIC $(CFLAGS)
$(OBJ_DIR)/device/Device.o: device/Device.cpp
gcc -c -o $@ $< $(INCLUDE) -fPIC $(CFLAGS)
$(OBJ_DIR)/%.o: %.c $(patsubst /%.o,/%.c,$@)
gcc -c -o $@ $< $(INCLUDE) -fPIC $(CFLAGS)
$(LIBS): $(OBJS)
gcc -shared -o $@ $^ $(LDFLAGS) -Wl,-rpath=$(LIB_PATH) -L$(LIB_PATH) -lthreads -lnervcore -lluaT -lpthread -lstdc++
ECHO:
@echo $(SUBMODULE)
clean:
-rm -rf $(OBJ_DIR)
|