X-Git-Url: https://git.ti.com/gitweb?p=jacinto-ai%2Fcaffe-jacinto.git;a=blobdiff_plain;f=Makefile;h=a74d8b51bcdf44f7585d9a38a8f3d3d9f5f1b07b;hp=99c4c4b89ad39f6db78556066f3b07229e72d13a;hb=82b912be849a7bec9bfee92a8e5d81182f4130f2;hpb=7149de51ceb87e204e3a50d41cb1495caab965dd

diff --git a/Makefile b/Makefile
index 99c4c4b8..a74d8b51 100644
--- a/Makefile
+++ b/Makefile
@@ -1,41 +1,111 @@
-#
-# The following defines a variable named "NAME" with a value of "myprogram". By convention,
-# a lowercase prefix (in this case "program") and an uppercased suffix (in this case "NAME"), separated
-# by an underscore is used to name attributes for a common element. Think of this like
-# using program.NAME, program.C_SRCS, etc. There are no structs in Make, so we use this convention
-# to keep track of attributes that all belong to the same target or program.  
-#
-NAME := caffeine.so
-C_SRCS := $(wildcard src/caffeine/*.c)
-CXX_SRCS := $(wildcard src/caffeine/*.cpp)
-C_OBJS := ${C_SRCS:.c=.o}
+# The makefile for caffe. Extremely hack.
+PROJECT := caffe
+TEST_GPUID := 1
+
+# The target static library and shared library name
+NAME := lib$(PROJECT).so
+STATIC_NAME := lib$(PROJECT).a
+# All source files
+CXX_SRCS := $(shell find src/caffe ! -name "test_*.cpp" -name "*.cpp")
+CU_SRCS := $(shell find src/caffe -name "*.cu")
+TEST_SRCS := $(shell find src/caffe -name "test_*.cpp")
+GTEST_SRC := src/gtest/gtest-all.cpp
+EXAMPLE_SRCS := $(shell find examples -name "*.cpp")
+PROTO_SRCS := $(wildcard src/caffe/proto/*.proto)
+# The generated files for protocol buffers
+PROTO_GEN_HEADER := ${PROTO_SRCS:.proto=.pb.h}
+PROTO_GEN_CC := ${PROTO_SRCS:.proto=.pb.cc}
+PROTO_GEN_PY := ${PROTO_SRCS:.proto=_pb2.py}
+# The objects that are needed to generate the library
 CXX_OBJS := ${CXX_SRCS:.cpp=.o}
-OBJS := $(C_OBJS) $(CXX_OBJS)
+CU_OBJS := ${CU_SRCS:.cu=.cuo}
+PROTO_OBJS := ${PROTO_GEN_CC:.cc=.o}
+OBJS := $(PROTO_OBJS) $(CXX_OBJS) $(CU_OBJS)
+# program and test objects
+EXAMPLE_OBJS := ${EXAMPLE_SRCS:.cpp=.o}
+TEST_OBJS := ${TEST_SRCS:.cpp=.o}
+GTEST_OBJ := ${GTEST_SRC:.cpp=.o}
+# program and test bins
+EXAMPLE_BINS :=${EXAMPLE_OBJS:.o=.bin}
+TEST_BINS := ${TEST_OBJS:.o=.testbin}
+
+# define third-party library paths
+CUDA_DIR := /usr/local/cuda
+CUDA_ARCH := -arch=sm_30
+MKL_DIR := /opt/intel/mkl
 
-CUDA_DIR = /usr/local/cuda
-CUDA_INCLUDE_DIR = $(CUDA_DIR)/include
-CUDA_LIB_DIR = $(CUDA_DIR)/lib
+CUDA_INCLUDE_DIR := $(CUDA_DIR)/include
+CUDA_LIB_DIR := $(CUDA_DIR)/lib64
+MKL_INCLUDE_DIR := $(MKL_DIR)/include
+MKL_LIB_DIR := $(MKL_DIR)/lib $(MKL_DIR)/lib/intel64
 
-INCLUDE_DIRS := $(CUDA_INCLUDE_DIR) src/
-LIBRARY_DIRS := $(CUDA_LIB_DIR)
-LIBRARIES := cudart cublas
+# define inclue and libaries
+# We put src here just for gtest
+INCLUDE_DIRS := ./src ./include /usr/local/include $(CUDA_INCLUDE_DIR) $(MKL_INCLUDE_DIR)
+LIBRARY_DIRS := /usr/lib /usr/local/lib $(CUDA_LIB_DIR) $(MKL_LIB_DIR)
+LIBRARIES := cuda cudart cublas curand protobuf opencv_core opencv_highgui \
+	glog mkl_rt mkl_intel_thread leveldb snappy pthread
 WARNINGS := -Wall
 
-CPPFLAGS += $(foreach includedir,$(INCLUDE_DIRS),-I$(includedir))
-LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir))
-LDFLAGS += $(foreach library,$(LIBRARIES),-l$(library)) -shared
+COMMON_FLAGS := -DNDEBUG $(foreach includedir,$(INCLUDE_DIRS),-I$(includedir))
+CXXFLAGS += -pthread -fPIC -O2 $(COMMON_FLAGS)
+NVCCFLAGS := -Xcompiler -fPIC -O2 $(COMMON_FLAGS)
+LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir)) \
+		$(foreach library,$(LIBRARIES),-l$(library))
+
+NVCC = nvcc $(NVCCFLAGS) $(CPPFLAGS) $(CUDA_ARCH)
+
+.PHONY: all test clean distclean linecount examples distribute
+
+all: $(NAME) $(STATIC_NAME) test examples
+
+linecount: clean
+	cloc --read-lang-def=caffe.cloc src/caffe/
+
+test: $(TEST_BINS)
 
-LINK = $(CXX) $(CXXFLAGS) $(CPPFLAGS) $(LDFLAGS) $(WARNINGS)
+examples: $(EXAMPLE_BINS)
 
-.PHONY: all clean distclean
+$(NAME): $(PROTO_OBJS) $(OBJS)
+	$(CXX) -shared $(OBJS) -o $(NAME) $(LDFLAGS) $(WARNINGS)
 
-all: $(NAME)
+$(STATIC_NAME): $(PROTO_OBJS) $(OBJS)
+	ar rcs $(STATIC_NAME) $(PROTO_OBJS) $(OBJS)
 
-$(NAME): $(OBJS)
-	$(LINK) $(OBJS) -o $(NAME)
+runtest: test
+	for testbin in $(TEST_BINS); do $$testbin $(TEST_GPUID); done
+
+$(TEST_BINS): %.testbin : %.o $(GTEST_OBJ) $(STATIC_NAME)
+	$(CXX) $< $(GTEST_OBJ) $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
+
+$(EXAMPLE_BINS): %.bin : %.o $(STATIC_NAME)
+	$(CXX) $< $(STATIC_NAME) -o $@ $(LDFLAGS) $(WARNINGS)
+
+$(OBJS): $(PROTO_GEN_CC)
+
+$(EXAMPLE_OBJS): $(PROTO_GEN_CC)
+
+$(CU_OBJS): %.cuo: %.cu
+	$(NVCC) -c $< -o $@
+
+$(PROTO_GEN_CC): $(PROTO_SRCS)
+	protoc --proto_path=src --cpp_out=src --python_out=src $(PROTO_SRCS)
+	mkdir -p include/caffe/proto
+	cp $(PROTO_GEN_HEADER) include/caffe/proto/
 
 clean:
-	@- $(RM) $(NAME)
-	@- $(RM) $(OBJS)
+	@- $(RM) $(NAME) $(STATIC_NAME) $(TEST_BINS) $(EXAMPLE_BINS)
+	@- $(RM) $(OBJS) $(TEST_OBJS) $(EXAMPLE_OBJS)
+	@- $(RM) $(PROTO_GEN_HEADER) $(PROTO_GEN_CC) $(PROTO_GEN_PY)
+	@- $(RM) -rf build
+
+distclean: clean
 
-distclean: clean
\ No newline at end of file
+distribute: all
+	mkdir build
+	cp -r include build/
+	mkdir build/bin
+	cp $(EXAMPLE_BINS) build/bin
+	mkdir build/lib
+	cp $(NAME) build/lib
+	cp $(STATIC_NAME) build/lib