1 # The makefile for caffe. Extremely hack.
2 PROJECT := caffe
3 TEST_GPUID := 0
5 # The target static library and shared library name
6 NAME := lib$(PROJECT).so
7 STATIC_NAME := lib$(PROJECT).a
8 # All source files
9 CXX_SRCS := $(shell find src/caffe ! -name "test_*.cpp" -name "*.cpp")
10 CU_SRCS := $(shell find src/caffe -name "*.cu")
11 TEST_SRCS := $(shell find src/caffe -name "test_*.cpp")
12 GTEST_SRC := src/gtest/gtest-all.cpp
13 EXAMPLE_SRCS := $(shell find examples -name "*.cpp")
14 PROTO_SRCS := $(wildcard src/caffe/proto/*.proto)
15 # The generated files for protocol buffers
16 PROTO_GEN_HEADER := ${PROTO_SRCS:.proto=.pb.h}
17 PROTO_GEN_CC := ${PROTO_SRCS:.proto=.pb.cc}
18 PROTO_GEN_PY := ${PROTO_SRCS:.proto=_pb2.py}
19 # The objects that are needed to generate the library
20 CXX_OBJS := ${CXX_SRCS:.cpp=.o}
21 CU_OBJS := ${CU_SRCS:.cu=.cuo}
22 PROTO_OBJS := ${PROTO_GEN_CC:.cc=.o}
23 OBJS := $(PROTO_OBJS) $(CXX_OBJS) $(CU_OBJS)
24 # program and test objects
25 EXAMPLE_OBJS := ${EXAMPLE_SRCS:.cpp=.o}
26 TEST_OBJS := ${TEST_SRCS:.cpp=.o}
27 GTEST_OBJ := ${GTEST_SRC:.cpp=.o}
28 # program and test bins
29 EXAMPLE_BINS :=${EXAMPLE_OBJS:.o=.bin}
30 TEST_BINS := ${TEST_OBJS:.o=.testbin}
32 # define third-party library paths
33 CUDA_DIR := /usr/local/cuda
34 CUDA_ARCH := -arch=sm_30
35 MKL_DIR := /opt/intel/mkl
37 CUDA_INCLUDE_DIR := $(CUDA_DIR)/include
38 CUDA_LIB_DIR := $(CUDA_DIR)/lib64
39 MKL_INCLUDE_DIR := $(MKL_DIR)/include
40 MKL_LIB_DIR := $(MKL_DIR)/lib $(MKL_DIR)/lib/intel64
42 # define inclue and libaries
43 # We put src here just for gtest
44 INCLUDE_DIRS := ./src ./include /usr/local/include $(CUDA_INCLUDE_DIR) \
45 $(MKL_INCLUDE_DIR)
46 LIBRARY_DIRS := /usr/lib /usr/local/lib $(CUDA_LIB_DIR) $(MKL_LIB_DIR)
47 LIBRARIES := cuda cudart cublas curand protobuf opencv_core opencv_highgui \
48 glog mkl_rt mkl_intel_thread leveldb snappy pthread boost_system \
49 opencv_imgproc
50 WARNINGS := -Wall
52 COMMON_FLAGS := -DNDEBUG $(foreach includedir,$(INCLUDE_DIRS),-I$(includedir))
53 CXXFLAGS += -pthread -fPIC -O2 $(COMMON_FLAGS)
54 NVCCFLAGS := -Xcompiler -fPIC -O2 $(COMMON_FLAGS)
55 LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir)) \
56 $(foreach library,$(LIBRARIES),-l$(library))
58 NVCC = nvcc $(NVCCFLAGS) $(CPPFLAGS) $(CUDA_ARCH)
60 .PHONY: all test clean distclean linecount examples pycaffe distribute
62 all: $(NAME) $(STATIC_NAME) test examples pycaffe
64 pycaffe: $(STATIC_NAME) python/caffe/pycaffe.cpp $(PROTO_GEN_PY)
65 protoc --proto_path=src --python_out=python $(PROTO_SRCS)
66 $(CXX) -o python/caffe/pycaffe.so -I/usr/include/python2.7 \
67 -I/usr/local/lib/python2.7/dist-packages/numpy/core/include/ -shared \
68 python/caffe/pycaffe.cpp $(STATIC_NAME) $(CXXFLAGS) $(LDFLAGS) \
69 $(WARNING) -lboost_python -lpython2.7
71 linecount: clean
72 cloc --read-lang-def=caffe.cloc src/caffe/
74 test: $(TEST_BINS)
76 examples: $(EXAMPLE_BINS)
78 $(NAME): $(PROTO_OBJS) $(OBJS)
79 $(CXX) -shared $(OBJS) -o $(NAME) $(LDFLAGS) $(WARNINGS)
81 $(STATIC_NAME): $(PROTO_OBJS) $(OBJS)
82 ar rcs $(STATIC_NAME) $(PROTO_OBJS) $(OBJS)
84 runtest: test
85 for testbin in $(TEST_BINS); do $$testbin $(TEST_GPUID); done
87 $(TEST_BINS): %.testbin : %.o $(GTEST_OBJ) $(STATIC_NAME)
88 $(CXX) $< $(GTEST_OBJ) $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
90 $(EXAMPLE_BINS): %.bin : %.o $(STATIC_NAME)
91 $(CXX) $< $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
93 $(OBJS): $(PROTO_GEN_CC)
95 $(EXAMPLE_OBJS): $(PROTO_GEN_CC)
97 $(CU_OBJS): %.cuo: %.cu
98 $(NVCC) -c $< -o $@
100 $(PROTO_GEN_CC): $(PROTO_SRCS)
101 protoc --proto_path=src --cpp_out=src --python_out=src $(PROTO_SRCS)
102 mkdir -p include/caffe/proto
103 cp $(PROTO_GEN_HEADER) include/caffe/proto/
105 clean:
106 @- $(RM) $(NAME) $(STATIC_NAME) $(TEST_BINS) $(EXAMPLE_BINS)
107 @- $(RM) $(OBJS) $(TEST_OBJS) $(EXAMPLE_OBJS)
108 @- $(RM) $(PROTO_GEN_HEADER) $(PROTO_GEN_CC) $(PROTO_GEN_PY)
109 @- $(RM) include/caffe/proto/caffe.pb.h
110 @- $(RM) python/caffe/proto/caffe_pb2.py
111 @- $(RM) -rf build
113 distclean: clean
115 distribute: all
116 mkdir build
117 cp -r include build/
118 mkdir build/bin
119 cp $(EXAMPLE_BINS) build/bin
120 mkdir build/lib
121 cp $(NAME) build/lib
122 cp $(STATIC_NAME) build/lib