src/Makefile

   1 # The makefile for caffe. Extremely hack.
   2 PROJECT := caffe
   3 TEST_GPUID := 1
   4
   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 caffe ! -name "test_*.cpp" -name "*.cpp")
  10 CU_SRCS := $(shell find caffe -name "*.cu")
  11 TEST_SRCS := $(shell find caffe -name "test_*.cpp")
  12 GTEST_SRC := gtest/gtest-all.cpp
  13 PROGRAM_SRCS := $(shell find programs -name "*.cpp")
  14 PROTO_SRCS := $(wildcard 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 PROGRAM_OBJS := ${PROGRAM_SRCS:.cpp=.o}
  26 TEST_OBJS := ${TEST_SRCS:.cpp=.o}
  27 GTEST_OBJ := ${GTEST_SRC:.cpp=.o}
  28 # program and test bins
  29 PROGRAM_BINS :=${PROGRAM_OBJS:.o=.bin}
  30 TEST_BINS := ${TEST_OBJS:.o=.testbin}
  31
  32 # define third-party library paths
  33 CUDA_DIR := /usr/local/cuda
  34 CUDA_ARCH := -arch=sm_30
  35 MKL_DIR := /opt/intel/mkl
  36
  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
  41
  42 # define inclue and libaries
  43 INCLUDE_DIRS := . /usr/local/include $(CUDA_INCLUDE_DIR) $(MKL_INCLUDE_DIR)
  44 LIBRARY_DIRS := . /usr/lib /usr/local/lib $(CUDA_LIB_DIR) $(MKL_LIB_DIR)
  45 LIBRARIES := cuda cudart cublas protobuf glog mkl_rt mkl_intel_thread curand \
  46                 leveldb snappy pthread tcmalloc
  47 WARNINGS := -Wall
  48
  49 COMMON_FLAGS := $(foreach includedir,$(INCLUDE_DIRS),-I$(includedir))
  50 CXXFLAGS += -pthread -fPIC -O2 $(COMMON_FLAGS)
  51 NVCCFLAGS := -Xcompiler -fPIC -O2 $(COMMON_FLAGS)
  52 LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir)) \
  53                 $(foreach library,$(LIBRARIES),-l$(library))
  54
  55 NVCC = nvcc $(NVCCFLAGS) $(CPPFLAGS) $(CUDA_ARCH)
  56
  57 .PHONY: all test clean distclean linecount program
  58
  59 all: $(NAME) $(STATIC_NAME) test program
  60
  61 linecount: clean
  62         cloc --read-lang-def=caffe.cloc caffe/
  63
  64 test: $(TEST_BINS)
  65
  66 program: $(PROGRAM_BINS)
  67
  68 $(NAME): $(PROTO_OBJS) $(OBJS)
  69         $(CXX) -shared $(OBJS) -o $(NAME) $(LDFLAGS) $(WARNINGS)
  70
  71 $(STATIC_NAME): $(PROTO_OBJS) $(OBJS)
  72         ar rcs $(STATIC_NAME) $(PROTO_OBJS) $(OBJS)
  73
  74 runtest: test
  75         for testbin in $(TEST_BINS); do $$testbin $(TEST_GPUID); done
  76
  77 $(TEST_BINS): %.testbin : %.o $(GTEST_OBJ) $(STATIC_NAME)
  78         $(CXX) $< $(GTEST_OBJ) $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
  79
  80 $(PROGRAM_BINS): %.bin : %.o $(STATIC_NAME)
  81         $(CXX) $< $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
  82
  83 $(OBJS): $(PROTO_GEN_CC)
  84
  85 $(PROGRAM_OBJS): $(PROTO_GEN_CC)
  86
  87 $(CU_OBJS): %.cuo: %.cu
  88         $(NVCC) -c $< -o $@
  89
  90 $(PROTO_GEN_CC): $(PROTO_SRCS)
  91         protoc $(PROTO_SRCS) --cpp_out=. --python_out=.
  92
  93 clean:
  94         @- $(RM) $(NAME) $(STATIC_NAME) $(TEST_BINS) $(PROGRAM_BINS)
  95         @- $(RM) $(OBJS) $(TEST_OBJS) $(PROGRAM_OBJS)
  96         @- $(RM) $(PROTO_GEN_HEADER) $(PROTO_GEN_CC) $(PROTO_GEN_PY)
  97
  98 distclean: clean