a bunch of updates. to be checked on durian. does not build.
[jacinto-ai/caffe-jacinto.git] / src / Makefile
1 #
2 # The following defines a variable named "NAME" with a value of "myprogram". By convention,
3 # a lowercase prefix (in this case "program") and an uppercased suffix (in this case "NAME"), separated
4 # by an underscore is used to name attributes for a common element. Think of this like
5 # using program.NAME, program.C_SRCS, etc. There are no structs in Make, so we use this convention
6 # to keep track of attributes that all belong to the same target or program.  
7 #
8 CXX := nvcc
10 PROJECT := caffeine
11 NAME := lib$(PROJECT).so
12 TEST_NAME := test_$(PROJECT)
13 CXX_SRCS := $(shell find . ! -name "test_*.cpp" -name "*.cpp")
14 TEST_SRCS := $(shell find . -name "test_*.cpp")
15 PROTO_SRCS := $(wildcard caffeine/proto/*.proto)
16 PROTO_GEN_HEADER := ${PROTO_SRCS:.proto=.pb.h}
17 PROTO_GEN_CC := ${PROTO_SRCS:.proto=.pb.cc}
18 CXX_OBJS := ${CXX_SRCS:.cpp=.o}
19 PROTO_OBJS := ${PROTO_SRCS:.proto=.pb.o}
20 OBJS := $(CXX_OBJS) $(PROTO_OBJS)
21 TEST_OBJS := ${TEST_SRCS:.cpp=.o}
23 CUDA_DIR = /usr/local/cuda
25 CUDA_INCLUDE_DIR = $(CUDA_DIR)/include
26 CUDA_LIB_DIR = $(CUDA_DIR)/lib
28 INCLUDE_DIRS := . $(CUDA_INCLUDE_DIR)
29 LIBRARY_DIRS := . $(CUDA_LIB_DIR)
30 LIBRARIES := cuda cudart cublas protobuf
31 WARNINGS := -Wall
33 CPPFLAGS += $(foreach includedir,$(INCLUDE_DIRS),-I$(includedir))
34 LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir))
35 LDFLAGS += $(foreach library,$(LIBRARIES),-l$(library))
37 LINK = $(CXX) $(CXXFLAGS) $(CPPFLAGS) $(LDFLAGS) $(WARNINGS)
39 .PHONY: all test clean distclean
41 all: $(NAME)
43 test: $(TEST_NAME)
44   
45 $(TEST_NAME): $(TEST_OBJS) $(OBJS)
46         $(LINK) -o $(TEST_NAME) -l$(PROJECT) $(CXX_SRCS) $(TEST_SRCS) gtest/gtest-all.cc
48 $(NAME): $(PROTO_GEN_CC) $(OBJS)
49         $(LINK) -shared $(OBJS) -o $(NAME)
51 $(PROTO_GEN_CC): $(PROTO_SRCS)
52         protoc $(PROTO_SRCS) --cpp_out=.
54 clean:
55         $(RM) $(NAME)
56         $(RM) $(OBJS)
57         $(RM) $(PROTO_GEN_HEADER) $(PROTO_GEN_CC)
59 distclean: clean