misc update

[jacinto-ai/caffe-jacinto.git] / src / Makefile

diff --git a/src/Makefile b/src/Makefile
index 9fff4b99a152f5ab57c283ea9de51b2595d56055..fd8a8abf7b3fc750fda7557f10fc74ecc7d692ff 100644 (file)
--- a/src/Makefile
+++ b/src/Makefile
@@ -14,6 +14,7 @@ TEST_SRCS := $(shell find caffeine -name "test_*.cpp") gtest/gtest-all.cpp
 PROTO_SRCS := $(wildcard caffeine/proto/*.proto)
 PROTO_GEN_HEADER := ${PROTO_SRCS:.proto=.pb.h}
 PROTO_GEN_CC := ${PROTO_SRCS:.proto=.pb.cc}
+PROTO_GEN_PY := ${PROTO_SRCS:.proto=_pb2.py}
 CXX_OBJS := ${CXX_SRCS:.cpp=.o}
 CU_OBJS := ${CU_SRCS:.cu=.o}
 PROTO_OBJS := ${PROTO_SRCS:.proto=.pb.o}
@@ -41,10 +42,13 @@ LDFLAGS += $(foreach library,$(LIBRARIES),-l$(library))
 LINK = $(CXX) $(CXXFLAGS) $(CPPFLAGS) $(LDFLAGS) $(WARNINGS)
 NVCC = nvcc ${CXXFLAGS:-fPIC=-Xcompiler -fPIC} $(CPPFLAGS) $(CUDA_ARCH)
 
-.PHONY: all test clean distclean
+.PHONY: all test clean distclean linecount
 
 all: $(NAME)
 
+linecount:
+       cloc --read-lang-def=caffeine.cloc caffeine/
+
 test: $(TEST_NAME)
   
 $(TEST_NAME): $(OBJS) $(TEST_OBJS)
@@ -62,6 +66,6 @@ $(PROTO_GEN_CC): $(PROTO_SRCS)
 clean:
        @- $(RM) $(NAME) $(TEST_NAME)
        @- $(RM) $(OBJS) $(TEST_OBJS)
-       @- $(RM) $(PROTO_GEN_HEADER) $(PROTO_GEN_CC)
+       @- $(RM) $(PROTO_GEN_HEADER) $(PROTO_GEN_CC) $(PROTO_GEN_PY)
 
 distclean: clean