1 # The makefile for caffe. Extremely hacky.
2 PROJECT := caffe
3 TEST_GPUID := 0
5 include Makefile.config
7 ##############################################################################
8 # After this line, things should happen automatically.
9 ##############################################################################
11 # The target static library and shared library name
12 NAME := lib$(PROJECT).so
13 STATIC_NAME := lib$(PROJECT).a
15 ##############################
16 # Get all source files
17 ##############################
18 # CXX_SRCS are the source files excluding the test ones.
19 CXX_SRCS := $(shell find src/$(PROJECT) ! -name "test_*.cpp" -name "*.cpp")
20 # HXX_SRCS are the header files
21 HXX_SRCS := $(shell find include/$(PROJECT) ! -name "*.hpp")
22 # CU_SRCS are the cuda source files
23 CU_SRCS := $(shell find src/$(PROJECT) -name "*.cu")
24 # TEST_SRCS are the test source files
25 TEST_SRCS := $(shell find src/$(PROJECT) -name "test_*.cpp")
26 GTEST_SRC := src/gtest/gtest-all.cpp
27 # TEST_HDRS are the test header files
28 TEST_HDRS := $(shell find src/$(PROJECT) -name "test_*.hpp")
29 # EXAMPLE_SRCS are the source files for the example binaries
30 EXAMPLE_SRCS := $(shell find examples -name "*.cpp")
31 # PROTO_SRCS are the protocol buffer definitions
32 PROTO_SRCS := $(wildcard src/$(PROJECT)/proto/*.proto)
33 # PY$(PROJECT)_SRC is the python wrapper for $(PROJECT)
34 PY$(PROJECT)_SRC := python/$(PROJECT)/py$(PROJECT).cpp
35 PY$(PROJECT)_SO := python/$(PROJECT)/py$(PROJECT).so
36 # MAT$(PROJECT)_SRC is the matlab wrapper for $(PROJECT)
37 MAT$(PROJECT)_SRC := matlab/$(PROJECT)/mat$(PROJECT).cpp
38 MAT$(PROJECT)_SO := matlab/$(PROJECT)/$(PROJECT)
40 ##############################
41 # Derive generated files
42 ##############################
43 # The generated files for protocol buffers
44 PROTO_GEN_HEADER := ${PROTO_SRCS:.proto=.pb.h}
45 PROTO_GEN_CC := ${PROTO_SRCS:.proto=.pb.cc}
46 PROTO_GEN_PY := ${PROTO_SRCS:.proto=_pb2.py}
47 # The objects corresponding to the source files
48 # These objects will be linked into the final shared library, so we
49 # exclude the test and example objects.
50 CXX_OBJS := $(addprefix $(BUILD_DIR)/, ${CXX_SRCS:.cpp=.o})
51 CU_OBJS := $(addprefix $(BUILD_DIR)/, ${CU_SRCS:.cu=.cuo})
52 PROTO_OBJS := $(addprefix $(BUILD_DIR)/, ${PROTO_GEN_CC:.cc=.o})
53 OBJS := $(PROTO_OBJS) $(CXX_OBJS) $(CU_OBJS)
54 # program and test objects
55 EXAMPLE_OBJS := $(addprefix $(BUILD_DIR)/, ${EXAMPLE_SRCS:.cpp=.o})
56 TEST_OBJS := $(addprefix $(BUILD_DIR)/, ${TEST_SRCS:.cpp=.o})
57 GTEST_OBJ := $(addprefix $(BUILD_DIR)/, ${GTEST_SRC:.cpp=.o})
58 # program and test bins
59 EXAMPLE_BINS := ${EXAMPLE_OBJS:.o=.bin}
60 TEST_BINS := ${TEST_OBJS:.o=.testbin}
62 ##############################
63 # Derive include and lib directories
64 ##############################
65 CUDA_INCLUDE_DIR := $(CUDA_DIR)/include
66 CUDA_LIB_DIR := $(CUDA_DIR)/lib64 $(CUDA_DIR)/lib
67 MKL_INCLUDE_DIR := $(MKL_DIR)/include
68 MKL_LIB_DIR := $(MKL_DIR)/lib $(MKL_DIR)/lib/intel64
70 INCLUDE_DIRS += ./src ./include $(CUDA_INCLUDE_DIR) $(MKL_INCLUDE_DIR)
71 LIBRARY_DIRS += $(CUDA_LIB_DIR) $(MKL_LIB_DIR)
72 LIBRARIES := cudart cublas curand mkl_rt pthread \
73 glog protobuf leveldb snappy boost_system \
74 opencv_core opencv_highgui opencv_imgproc
75 PYTHON_LIBRARIES := boost_python python2.7
76 WARNINGS := -Wall
78 COMMON_FLAGS := -DNDEBUG -O2 $(foreach includedir,$(INCLUDE_DIRS),-I$(includedir))
79 CXXFLAGS += -pthread -fPIC $(COMMON_FLAGS)
80 NVCCFLAGS := -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
81 LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir)) \
82 $(foreach library,$(LIBRARIES),-l$(library))
83 PYTHON_LDFLAGS := $(LDFLAGS) $(foreach library,$(PYTHON_LIBRARIES),-l$(library))
86 ##############################
87 # Define build targets
88 ##############################
89 .PHONY: all init test clean linecount examples py mat distribute py$(PROJECT) mat$(PROJECT) proto
91 all: init $(NAME) $(STATIC_NAME) examples
92 @echo $(CXX_OBJS)
94 init:
95 @ mkdir -p $(foreach obj,$(OBJS),$(dir $(obj)))
96 @ mkdir -p $(foreach obj,$(EXAMPLE_OBJS),$(dir $(obj)))
97 @ mkdir -p $(foreach obj,$(TEST_OBJS),$(dir $(obj)))
98 @ mkdir -p $(foreach obj,$(GTEST_OBJ),$(dir $(obj)))
100 linecount: clean
101 cloc --read-lang-def=$(PROJECT).cloc src/$(PROJECT)/
103 test: init $(TEST_BINS)
105 examples: init $(EXAMPLE_BINS)
107 py$(PROJECT): py
109 py: init $(STATIC_NAME) $(PY$(PROJECT)_SRC) $(PROTO_GEN_PY)
110 $(CXX) -shared -o $(PY$(PROJECT)_SO) $(PY$(PROJECT)_SRC) \
111 $(STATIC_NAME) $(CXXFLAGS) $(PYTHON_LDFLAGS)
112 @echo
114 mat$(PROJECT): mat
116 mat: init $(STATIC_NAME) $(MAT$(PROJECT)_SRC)
117 $(MATLAB_DIR)/bin/mex $(MAT$(PROJECT)_SRC) $(STATIC_NAME) \
118 CXXFLAGS="\$$CXXFLAGS $(CXXFLAGS) $(WARNINGS)" \
119 CXXLIBS="\$$CXXLIBS $(LDFLAGS)" \
120 -o $(MAT$(PROJECT)_SO)
121 @echo
123 $(NAME): init $(PROTO_OBJS) $(OBJS)
124 $(CXX) -shared -o $(NAME) $(OBJS) $(LDFLAGS) $(WARNINGS)
125 @echo
127 $(STATIC_NAME): init $(PROTO_OBJS) $(OBJS)
128 ar rcs $(STATIC_NAME) $(PROTO_OBJS) $(OBJS)
129 @echo
131 runtest: test
132 for testbin in $(TEST_BINS); do $$testbin $(TEST_GPUID); done
134 $(TEST_BINS): %.testbin : %.o $(GTEST_OBJ) $(STATIC_NAME) $(TEST_HDRS)
135 $(CXX) $< $(GTEST_OBJ) $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
137 $(EXAMPLE_BINS): %.bin : %.o $(STATIC_NAME)
138 $(CXX) $< $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
139 @echo
141 $(OBJS): $(PROTO_GEN_CC) $(HXX_SRCS)
143 $(BUILD_DIR)/src/$(PROJECT)/%.o: src/$(PROJECT)/%.cpp
144 $(CXX) $< $(CXXFLAGS) -c -o $@ $(LDFLAGS)
145 @echo
147 $(BUILD_DIR)/src/$(PROJECT)/layers/%.o: src/$(PROJECT)/layers/%.cpp
148 $(CXX) $< $(CXXFLAGS) -c -o $@ $(LDFLAGS)
149 @echo
151 $(BUILD_DIR)/src/$(PROJECT)/proto/%.o: src/$(PROJECT)/proto/%.cc
152 $(CXX) $< $(CXXFLAGS) -c -o $@ $(LDFLAGS)
153 @echo
155 $(BUILD_DIR)/src/$(PROJECT)/test/%.o: src/test/%.cpp
156 $(CXX) $< $(CXXFLAGS) -c -o $@ $(LDFLAGS)
157 @echo
159 $(BUILD_DIR)/src/$(PROJECT)/util/%.o: src/$(PROJECT)/util/%.cpp
160 $(CXX) $< $(CXXFLAGS) -c -o $@ $(LDFLAGS)
161 @echo
163 $(BUILD_DIR)/src/gtest/%.o: src/gtest/%.cpp
164 $(CXX) $< $(CXXFLAGS) -c -o $@ $(LDFLAGS)
165 @echo
167 $(BUILD_DIR)/src/$(PROJECT)/layers/%.cuo: src/$(PROJECT)/layers/%.cu
168 $(CUDA_DIR)/bin/nvcc $(NVCCFLAGS) $(CUDA_ARCH) -c $< -o $@
169 @echo
171 $(BUILD_DIR)/src/$(PROJECT)/util/%.cuo: src/$(PROJECT)/util/%.cu
172 $(CUDA_DIR)/bin/nvcc $(NVCCFLAGS) $(CUDA_ARCH) -c $< -o $@
173 @echo
175 $(BUILD_DIR)/examples/%.o: examples/%.cpp
176 $(CXX) $< $(CXXFLAGS) -c -o $@ $(LDFLAGS)
177 @echo
179 $(PROTO_GEN_PY): $(PROTO_SRCS)
180 protoc --proto_path=src --python_out=python $(PROTO_SRCS)
181 @echo
183 proto: $(PROTO_GEN_CC)
185 $(PROTO_GEN_CC): $(PROTO_SRCS)
186 protoc --proto_path=src --cpp_out=src $(PROTO_SRCS)
187 mkdir -p include/$(PROJECT)/proto
188 cp $(PROTO_GEN_HEADER) include/$(PROJECT)/proto/
189 @echo
191 clean:
192 @- $(RM) $(NAME) $(STATIC_NAME)
193 @- $(RM) $(PROTO_GEN_HEADER) $(PROTO_GEN_CC) $(PROTO_GEN_PY)
194 @- $(RM) include/$(PROJECT)/proto/$(PROJECT).pb.h
195 @- $(RM) python/$(PROJECT)/proto/$(PROJECT)_pb2.py
196 @- $(RM) python/$(PROJECT)/*.so
197 @- $(RM) -rf $(BUILD_DIR)
198 @- $(RM) -rf $(DISTRIBUTE_DIR)
200 distribute: all
201 mkdir $(DISTRIBUTE_DIR)
202 # add include
203 cp -r include $(DISTRIBUTE_DIR)/
204 # add example binaries
205 mkdir $(DISTRIBUTE_DIR)/bin
206 cp $(EXAMPLE_BINS) $(DISTRIBUTE_DIR)/bin
207 # add libraries
208 mkdir $(DISTRIBUTE_DIR)/lib
209 cp $(NAME) $(DISTRIBUTE_DIR)/lib
210 cp $(STATIC_NAME) $(DISTRIBUTE_DIR)/lib
211 # add python - it's not the standard way, indeed...
212 cp -r python $(DISTRIBUTE_DIR)/python