point caffe url to bvlc
[jacinto-ai/caffe-jacinto.git] / Makefile
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/caffe ! -name "test_*.cpp" -name "*.cpp")
20 # HXX_SRCS are the header files
21 HXX_SRCS := $(shell find include/caffe ! -name "*.hpp")
22 # CU_SRCS are the cuda source files
23 CU_SRCS := $(shell find src/caffe -name "*.cu")
24 # TEST_SRCS are the test source files
25 TEST_SRCS := $(shell find src/caffe -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/caffe -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/caffe/proto/*.proto)
33 # PYCAFFE_SRC is the python wrapper for caffe
34 PYCAFFE_SRC := python/caffe/pycaffe.cpp
35 PYCAFFE_SO := python/caffe/pycaffe.so
36 # MATCAFFE_SRC is the matlab wrapper for caffe
37 MATCAFFE_SRC := matlab/caffe/matcaffe.cpp
38 MATCAFFE_SO := matlab/caffe/caffe
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 := ${CXX_SRCS:.cpp=.o}
51 CU_OBJS := ${CU_SRCS:.cu=.cuo}
52 PROTO_OBJS := ${PROTO_GEN_CC:.cc=.o}
53 OBJS := $(PROTO_OBJS) $(CXX_OBJS) $(CU_OBJS)
54 # program and test objects
55 EXAMPLE_OBJS := ${EXAMPLE_SRCS:.cpp=.o}
56 TEST_OBJS := ${TEST_SRCS:.cpp=.o}
57 GTEST_OBJ := ${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 protobuf opencv_core opencv_highgui \
73         glog mkl_rt mkl_intel_thread leveldb snappy pthread boost_system \
74         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 test clean linecount examples pycaffe distribute
91 all: $(NAME) $(STATIC_NAME) examples
93 linecount: clean
94         cloc --read-lang-def=caffe.cloc src/caffe/
96 test: $(TEST_BINS)
98 examples: $(EXAMPLE_BINS)
100 pycaffe: $(STATIC_NAME) $(PYCAFFE_SRC) $(PROTO_GEN_PY)
101         $(CXX) -shared -o $(PYCAFFE_SO) $(PYCAFFE_SRC) \
102                 $(STATIC_NAME) $(CXXFLAGS) $(PYTHON_LDFLAGS)
104 matcaffe: $(STATIC_NAME) $(MATCAFFE_SRC)
105         $(MATLAB_DIR)/bin/mex $(MATCAFFE_SRC) $(STATIC_NAME) \
106                 CXXFLAGS="\$$CXXFLAGS $(CXXFLAGS) $(WARNINGS)" \
107                 CXXLIBS="\$$CXXLIBS $(LDFLAGS)" \
108                 -o $(MATCAFFE_SO)
110 $(NAME): $(PROTO_OBJS) $(OBJS)
111         $(CXX) -shared -o $(NAME) $(OBJS) $(LDFLAGS) $(WARNINGS)
113 $(STATIC_NAME): $(PROTO_OBJS) $(OBJS)
114         ar rcs $(STATIC_NAME) $(PROTO_OBJS) $(OBJS)
116 runtest: test
117         for testbin in $(TEST_BINS); do $$testbin $(TEST_GPUID); done
119 $(TEST_BINS): %.testbin : %.o $(GTEST_OBJ) $(STATIC_NAME) $(TEST_HDRS)
120         $(CXX) $< $(GTEST_OBJ) $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
122 $(EXAMPLE_BINS): %.bin : %.o $(STATIC_NAME)
123         $(CXX) $< $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
125 $(OBJS): $(PROTO_GEN_CC) $(HXX_SRCS)
127 $(EXAMPLE_OBJS): $(PROTO_GEN_CC)
129 $(CU_OBJS): %.cuo: %.cu
130         $(CUDA_DIR)/bin/nvcc $(NVCCFLAGS) $(CUDA_ARCH) -c $< -o $@
132 $(PROTO_GEN_PY): $(PROTO_SRCS)
133         protoc --proto_path=src --python_out=python $(PROTO_SRCS)
135 $(PROTO_GEN_CC): $(PROTO_SRCS)
136         protoc --proto_path=src --cpp_out=src $(PROTO_SRCS)
137         mkdir -p include/caffe/proto
138         cp $(PROTO_GEN_HEADER) include/caffe/proto/
140 clean:
141         @- $(RM) $(NAME) $(STATIC_NAME) $(TEST_BINS) $(EXAMPLE_BINS)
142         @- $(RM) $(OBJS) $(TEST_OBJS) $(EXAMPLE_OBJS)
143         @- $(RM) $(PROTO_GEN_HEADER) $(PROTO_GEN_CC) $(PROTO_GEN_PY)
144         @- $(RM) include/caffe/proto/caffe.pb.h
145         @- $(RM) python/caffe/proto/caffe_pb2.py
146         @- $(RM) -rf build
148 distribute: all
149         mkdir build
150         # add include
151         cp -r include build/
152         # add example binaries
153         mkdir build/bin
154         cp $(EXAMPLE_BINS) build/bin
155         # add libraries
156         mkdir build/lib
157         cp $(NAME) build/lib
158         cp $(STATIC_NAME) build/lib
159         # add python - it's not the standard way, indeed...
160         cp -r python build/python