Makefile

   1 PROJECT := caffe
   2
   3 CONFIG_FILE := Makefile.config
   4 # Explicitly check for the config file, otherwise make -k will proceed anyway.
   5 ifeq ($(wildcard $(CONFIG_FILE)),)
   6 $(error $(CONFIG_FILE) not found. See $(CONFIG_FILE).example.)
   7 endif
   8 include $(CONFIG_FILE)
   9
  10 BUILD_DIR_LINK := $(BUILD_DIR)
  11 ifeq ($(RELEASE_BUILD_DIR),)
  12         RELEASE_BUILD_DIR := .$(BUILD_DIR)_release
  13 endif
  14 ifeq ($(DEBUG_BUILD_DIR),)
  15         DEBUG_BUILD_DIR := .$(BUILD_DIR)_debug
  16 endif
  17
  18 DEBUG ?= 0
  19 ifeq ($(DEBUG), 1)
  20         BUILD_DIR := $(DEBUG_BUILD_DIR)
  21         OTHER_BUILD_DIR := $(RELEASE_BUILD_DIR)
  22 else
  23         BUILD_DIR := $(RELEASE_BUILD_DIR)
  24         OTHER_BUILD_DIR := $(DEBUG_BUILD_DIR)
  25 endif
  26
  27 # All of the directories containing code.
  28 SRC_DIRS := $(shell find * -type d -exec bash -c "find {} -maxdepth 1 \
  29         \( -name '*.cpp' -o -name '*.proto' \) | grep -q ." \; -print)
  30
  31 # The target shared library name
  32 LIBRARY_NAME := $(PROJECT)
  33 LIB_BUILD_DIR := $(BUILD_DIR)/lib
  34 STATIC_NAME := $(LIB_BUILD_DIR)/lib$(LIBRARY_NAME).a
  35 DYNAMIC_VERSION_MAJOR           := 1
  36 DYNAMIC_VERSION_MINOR           := 0
  37 DYNAMIC_VERSION_REVISION        := 0-rc3
  38 DYNAMIC_NAME_SHORT := lib$(LIBRARY_NAME).so
  39 #DYNAMIC_SONAME_SHORT := $(DYNAMIC_NAME_SHORT).$(DYNAMIC_VERSION_MAJOR)
  40 DYNAMIC_VERSIONED_NAME_SHORT := $(DYNAMIC_NAME_SHORT).$(DYNAMIC_VERSION_MAJOR).$(DYNAMIC_VERSION_MINOR).$(DYNAMIC_VERSION_REVISION)
  41 DYNAMIC_NAME := $(LIB_BUILD_DIR)/$(DYNAMIC_VERSIONED_NAME_SHORT)
  42 COMMON_FLAGS += -DCAFFE_VERSION=$(DYNAMIC_VERSION_MAJOR).$(DYNAMIC_VERSION_MINOR).$(DYNAMIC_VERSION_REVISION)
  43
  44 ##############################
  45 # Get all source files
  46 ##############################
  47 # CXX_SRCS are the source files excluding the test ones.
  48 CXX_SRCS := $(shell find src/$(PROJECT) ! -name "test_*.cpp" -name "*.cpp")
  49 # CU_SRCS are the cuda source files
  50 CU_SRCS := $(shell find src/$(PROJECT) ! -name "test_*.cu" -name "*.cu")
  51 # TEST_SRCS are the test source files
  52 TEST_MAIN_SRC := src/$(PROJECT)/test/test_caffe_main.cpp
  53 TEST_SRCS := $(shell find src/$(PROJECT) -name "test_*.cpp")
  54 TEST_SRCS := $(filter-out $(TEST_MAIN_SRC), $(TEST_SRCS))
  55 TEST_CU_SRCS := $(shell find src/$(PROJECT) -name "test_*.cu")
  56 GTEST_SRC := src/gtest/gtest-all.cpp
  57 # TOOL_SRCS are the source files for the tool binaries
  58 TOOL_SRCS := $(shell find tools -name "*.cpp")
  59 # EXAMPLE_SRCS are the source files for the example binaries
  60 EXAMPLE_SRCS := $(shell find examples -name "*.cpp")
  61 # BUILD_INCLUDE_DIR contains any generated header files we want to include.
  62 BUILD_INCLUDE_DIR := $(BUILD_DIR)/src
  63 # PROTO_SRCS are the protocol buffer definitions
  64 PROTO_SRC_DIR := src/$(PROJECT)/proto
  65 PROTO_SRCS := $(wildcard $(PROTO_SRC_DIR)/*.proto)
  66 # PROTO_BUILD_DIR will contain the .cc and obj files generated from
  67 # PROTO_SRCS; PROTO_BUILD_INCLUDE_DIR will contain the .h header files
  68 PROTO_BUILD_DIR := $(BUILD_DIR)/$(PROTO_SRC_DIR)
  69 PROTO_BUILD_INCLUDE_DIR := $(BUILD_INCLUDE_DIR)/$(PROJECT)/proto
  70 # NONGEN_CXX_SRCS includes all source/header files except those generated
  71 # automatically (e.g., by proto).
  72 NONGEN_CXX_SRCS := $(shell find \
  73         src/$(PROJECT) \
  74         include/$(PROJECT) \
  75         python/$(PROJECT) \
  76         matlab/+$(PROJECT)/private \
  77         examples \
  78         tools \
  79         -name "*.cpp" -or -name "*.hpp" -or -name "*.cu" -or -name "*.cuh")
  80 LINT_SCRIPT := scripts/cpp_lint.py
  81 LINT_OUTPUT_DIR := $(BUILD_DIR)/.lint
  82 LINT_EXT := lint.txt
  83 LINT_OUTPUTS := $(addsuffix .$(LINT_EXT), $(addprefix $(LINT_OUTPUT_DIR)/, $(NONGEN_CXX_SRCS)))
  84 EMPTY_LINT_REPORT := $(BUILD_DIR)/.$(LINT_EXT)
  85 NONEMPTY_LINT_REPORT := $(BUILD_DIR)/$(LINT_EXT)
  86 # PY$(PROJECT)_SRC is the python wrapper for $(PROJECT)
  87 PY$(PROJECT)_SRC := python/$(PROJECT)/_$(PROJECT).cpp
  88 PY$(PROJECT)_SO := python/$(PROJECT)/_$(PROJECT).so
  89 PY$(PROJECT)_HXX := include/$(PROJECT)/layers/python_layer.hpp
  90 # MAT$(PROJECT)_SRC is the mex entrance point of matlab package for $(PROJECT)
  91 MAT$(PROJECT)_SRC := matlab/+$(PROJECT)/private/$(PROJECT)_.cpp
  92 ifneq ($(MATLAB_DIR),)
  93         MAT_SO_EXT := $(shell $(MATLAB_DIR)/bin/mexext)
  94 endif
  95 MAT$(PROJECT)_SO := matlab/+$(PROJECT)/private/$(PROJECT)_.$(MAT_SO_EXT)
  96
  97 ##############################
  98 # Derive generated files
  99 ##############################
 100 # The generated files for protocol buffers
 101 PROTO_GEN_HEADER_SRCS := $(addprefix $(PROTO_BUILD_DIR)/, \
 102                 $(notdir ${PROTO_SRCS:.proto=.pb.h}))
 103 PROTO_GEN_HEADER := $(addprefix $(PROTO_BUILD_INCLUDE_DIR)/, \
 104                 $(notdir ${PROTO_SRCS:.proto=.pb.h}))
 105 PROTO_GEN_CC := $(addprefix $(BUILD_DIR)/, ${PROTO_SRCS:.proto=.pb.cc})
 106 PY_PROTO_BUILD_DIR := python/$(PROJECT)/proto
 107 PY_PROTO_INIT := python/$(PROJECT)/proto/__init__.py
 108 PROTO_GEN_PY := $(foreach file,${PROTO_SRCS:.proto=_pb2.py}, \
 109                 $(PY_PROTO_BUILD_DIR)/$(notdir $(file)))
 110 # The objects corresponding to the source files
 111 # These objects will be linked into the final shared library, so we
 112 # exclude the tool, example, and test objects.
 113 CXX_OBJS := $(addprefix $(BUILD_DIR)/, ${CXX_SRCS:.cpp=.o})
 114 CU_OBJS := $(addprefix $(BUILD_DIR)/cuda/, ${CU_SRCS:.cu=.o})
 115 PROTO_OBJS := ${PROTO_GEN_CC:.cc=.o}
 116 OBJS := $(PROTO_OBJS) $(CXX_OBJS) $(CU_OBJS)
 117 # tool, example, and test objects
 118 TOOL_OBJS := $(addprefix $(BUILD_DIR)/, ${TOOL_SRCS:.cpp=.o})
 119 TOOL_BUILD_DIR := $(BUILD_DIR)/tools
 120 TEST_CXX_BUILD_DIR := $(BUILD_DIR)/src/$(PROJECT)/test
 121 TEST_CU_BUILD_DIR := $(BUILD_DIR)/cuda/src/$(PROJECT)/test
 122 TEST_CXX_OBJS := $(addprefix $(BUILD_DIR)/, ${TEST_SRCS:.cpp=.o})
 123 TEST_CU_OBJS := $(addprefix $(BUILD_DIR)/cuda/, ${TEST_CU_SRCS:.cu=.o})
 124 TEST_OBJS := $(TEST_CXX_OBJS) $(TEST_CU_OBJS)
 125 GTEST_OBJ := $(addprefix $(BUILD_DIR)/, ${GTEST_SRC:.cpp=.o})
 126 EXAMPLE_OBJS := $(addprefix $(BUILD_DIR)/, ${EXAMPLE_SRCS:.cpp=.o})
 127 # Output files for automatic dependency generation
 128 DEPS := ${CXX_OBJS:.o=.d} ${CU_OBJS:.o=.d} ${TEST_CXX_OBJS:.o=.d} \
 129         ${TEST_CU_OBJS:.o=.d} $(BUILD_DIR)/${MAT$(PROJECT)_SO:.$(MAT_SO_EXT)=.d}
 130 # tool, example, and test bins
 131 TOOL_BINS := ${TOOL_OBJS:.o=.bin}
 132 EXAMPLE_BINS := ${EXAMPLE_OBJS:.o=.bin}
 133 # symlinks to tool bins without the ".bin" extension
 134 TOOL_BIN_LINKS := ${TOOL_BINS:.bin=}
 135 # Put the test binaries in build/test for convenience.
 136 TEST_BIN_DIR := $(BUILD_DIR)/test
 137 TEST_CU_BINS := $(addsuffix .testbin,$(addprefix $(TEST_BIN_DIR)/, \
 138                 $(foreach obj,$(TEST_CU_OBJS),$(basename $(notdir $(obj))))))
 139 TEST_CXX_BINS := $(addsuffix .testbin,$(addprefix $(TEST_BIN_DIR)/, \
 140                 $(foreach obj,$(TEST_CXX_OBJS),$(basename $(notdir $(obj))))))
 141 TEST_BINS := $(TEST_CXX_BINS) $(TEST_CU_BINS)
 142 # TEST_ALL_BIN is the test binary that links caffe dynamically.
 143 TEST_ALL_BIN := $(TEST_BIN_DIR)/test_all.testbin
 144
 145 ##############################
 146 # Derive compiler warning dump locations
 147 ##############################
 148 WARNS_EXT := warnings.txt
 149 CXX_WARNS := $(addprefix $(BUILD_DIR)/, ${CXX_SRCS:.cpp=.o.$(WARNS_EXT)})
 150 CU_WARNS := $(addprefix $(BUILD_DIR)/cuda/, ${CU_SRCS:.cu=.o.$(WARNS_EXT)})
 151 TOOL_WARNS := $(addprefix $(BUILD_DIR)/, ${TOOL_SRCS:.cpp=.o.$(WARNS_EXT)})
 152 EXAMPLE_WARNS := $(addprefix $(BUILD_DIR)/, ${EXAMPLE_SRCS:.cpp=.o.$(WARNS_EXT)})
 153 TEST_WARNS := $(addprefix $(BUILD_DIR)/, ${TEST_SRCS:.cpp=.o.$(WARNS_EXT)})
 154 TEST_CU_WARNS := $(addprefix $(BUILD_DIR)/cuda/, ${TEST_CU_SRCS:.cu=.o.$(WARNS_EXT)})
 155 ALL_CXX_WARNS := $(CXX_WARNS) $(TOOL_WARNS) $(EXAMPLE_WARNS) $(TEST_WARNS)
 156 ALL_CU_WARNS := $(CU_WARNS) $(TEST_CU_WARNS)
 157 ALL_WARNS := $(ALL_CXX_WARNS) $(ALL_CU_WARNS)
 158
 159 EMPTY_WARN_REPORT := $(BUILD_DIR)/.$(WARNS_EXT)
 160 NONEMPTY_WARN_REPORT := $(BUILD_DIR)/$(WARNS_EXT)
 161
 162 ##############################
 163 # Derive include and lib directories
 164 ##############################
 165 CUDA_INCLUDE_DIR := $(CUDA_DIR)/include
 166
 167 CUDA_LIB_DIR :=
 168 # add <cuda>/lib64 only if it exists
 169 ifneq ("$(wildcard $(CUDA_DIR)/lib64)","")
 170         CUDA_LIB_DIR += $(CUDA_DIR)/lib64
 171 endif
 172 CUDA_LIB_DIR += $(CUDA_DIR)/lib
 173
 174 INCLUDE_DIRS += $(BUILD_INCLUDE_DIR) ./src ./include
 175 ifneq ($(CPU_ONLY), 1)
 176         INCLUDE_DIRS += $(CUDA_INCLUDE_DIR)
 177         LIBRARY_DIRS += $(CUDA_LIB_DIR)
 178         LIBRARIES := cudart cublas curand
 179 endif
 180
 181 LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
 182
 183 # handle IO dependencies
 184 USE_LEVELDB ?= 1
 185 USE_LMDB ?= 1
 186 USE_OPENCV ?= 1
 187
 188 ifeq ($(USE_LEVELDB), 1)
 189         LIBRARIES += leveldb snappy
 190 endif
 191 ifeq ($(USE_LMDB), 1)
 192         LIBRARIES += lmdb
 193 endif
 194 ifeq ($(USE_OPENCV), 1)
 195         LIBRARIES += opencv_core opencv_highgui opencv_imgproc
 196
 197         ifeq ($(OPENCV_VERSION), 3)
 198                 LIBRARIES += opencv_imgcodecs
 199         endif
 200
 201 endif
 202 PYTHON_LIBRARIES ?= boost_python python2.7
 203 WARNINGS := -Wall -Wno-sign-compare
 204
 205 ##############################
 206 # Set build directories
 207 ##############################
 208
 209 DISTRIBUTE_DIR ?= distribute
 210 DISTRIBUTE_SUBDIRS := $(DISTRIBUTE_DIR)/bin $(DISTRIBUTE_DIR)/lib
 211 DIST_ALIASES := dist
 212 ifneq ($(strip $(DISTRIBUTE_DIR)),distribute)
 213                 DIST_ALIASES += distribute
 214 endif
 215
 216 ALL_BUILD_DIRS := $(sort $(BUILD_DIR) $(addprefix $(BUILD_DIR)/, $(SRC_DIRS)) \
 217         $(addprefix $(BUILD_DIR)/cuda/, $(SRC_DIRS)) \
 218         $(LIB_BUILD_DIR) $(TEST_BIN_DIR) $(PY_PROTO_BUILD_DIR) $(LINT_OUTPUT_DIR) \
 219         $(DISTRIBUTE_SUBDIRS) $(PROTO_BUILD_INCLUDE_DIR))
 220
 221 ##############################
 222 # Set directory for Doxygen-generated documentation
 223 ##############################
 224 DOXYGEN_CONFIG_FILE ?= ./.Doxyfile
 225 # should be the same as OUTPUT_DIRECTORY in the .Doxyfile
 226 DOXYGEN_OUTPUT_DIR ?= ./doxygen
 227 DOXYGEN_COMMAND ?= doxygen
 228 # All the files that might have Doxygen documentation.
 229 DOXYGEN_SOURCES := $(shell find \
 230         src/$(PROJECT) \
 231         include/$(PROJECT) \
 232         python/ \
 233         matlab/ \
 234         examples \
 235         tools \
 236         -name "*.cpp" -or -name "*.hpp" -or -name "*.cu" -or -name "*.cuh" -or \
 237         -name "*.py" -or -name "*.m")
 238 DOXYGEN_SOURCES += $(DOXYGEN_CONFIG_FILE)
 239
 240
 241 ##############################
 242 # Configure build
 243 ##############################
 244
 245 # Determine platform
 246 UNAME := $(shell uname -s)
 247 ifeq ($(UNAME), Linux)
 248         LINUX := 1
 249 else ifeq ($(UNAME), Darwin)
 250         OSX := 1
 251 endif
 252
 253 # Linux
 254 ifeq ($(LINUX), 1)
 255         CXX ?= /usr/bin/g++
 256         GCCVERSION := $(shell $(CXX) -dumpversion | cut -f1,2 -d.)
 257         # older versions of gcc are too dumb to build boost with -Wuninitalized
 258         ifeq ($(shell echo | awk '{exit $(GCCVERSION) < 4.6;}'), 1)
 259                 WARNINGS += -Wno-uninitialized
 260         endif
 261         # boost::thread is reasonably called boost_thread (compare OS X)
 262         # We will also explicitly add stdc++ to the link target.
 263         LIBRARIES += boost_thread stdc++
 264         VERSIONFLAGS += -Wl,-soname,$(DYNAMIC_VERSIONED_NAME_SHORT) -Wl,-rpath,$(ORIGIN)/../lib
 265 endif
 266
 267 # OS X:
 268 # clang++ instead of g++
 269 # libstdc++ for NVCC compatibility on OS X >= 10.9 with CUDA < 7.0
 270 ifeq ($(OSX), 1)
 271         CXX := /usr/bin/clang++
 272         ifneq ($(CPU_ONLY), 1)
 273                 CUDA_VERSION := $(shell $(CUDA_DIR)/bin/nvcc -V | grep -o 'release \d' | grep -o '\d')
 274                 ifeq ($(shell echo | awk '{exit $(CUDA_VERSION) < 7.0;}'), 1)
 275                         CXXFLAGS += -stdlib=libstdc++
 276                         LINKFLAGS += -stdlib=libstdc++
 277                 endif
 278                 # clang throws this warning for cuda headers
 279                 WARNINGS += -Wno-unneeded-internal-declaration
 280         endif
 281         # gtest needs to use its own tuple to not conflict with clang
 282         COMMON_FLAGS += -DGTEST_USE_OWN_TR1_TUPLE=1
 283         # boost::thread is called boost_thread-mt to mark multithreading on OS X
 284         LIBRARIES += boost_thread-mt
 285         # we need to explicitly ask for the rpath to be obeyed
 286         ORIGIN := @loader_path
 287         VERSIONFLAGS += -Wl,-install_name,@rpath/$(DYNAMIC_VERSIONED_NAME_SHORT) -Wl,-rpath,$(ORIGIN)/../../build/lib
 288 else
 289         ORIGIN := \$$ORIGIN
 290 endif
 291
 292 # Custom compiler
 293 ifdef CUSTOM_CXX
 294         CXX := $(CUSTOM_CXX)
 295 endif
 296
 297 # Static linking
 298 ifneq (,$(findstring clang++,$(CXX)))
 299         STATIC_LINK_COMMAND := -Wl,-force_load $(STATIC_NAME)
 300 else ifneq (,$(findstring g++,$(CXX)))
 301         STATIC_LINK_COMMAND := -Wl,--whole-archive $(STATIC_NAME) -Wl,--no-whole-archive
 302 else
 303   # The following line must not be indented with a tab, since we are not inside a target
 304   $(error Cannot static link with the $(CXX) compiler)
 305 endif
 306
 307 # Debugging
 308 ifeq ($(DEBUG), 1)
 309         COMMON_FLAGS += -DDEBUG -g -O0
 310         NVCCFLAGS += -G
 311 else
 312         COMMON_FLAGS += -DNDEBUG -O2
 313 endif
 314
 315 # cuDNN acceleration configuration.
 316 ifeq ($(USE_CUDNN), 1)
 317         LIBRARIES += cudnn
 318         COMMON_FLAGS += -DUSE_CUDNN
 319 endif
 320
 321 # configure IO libraries
 322 ifeq ($(USE_OPENCV), 1)
 323         COMMON_FLAGS += -DUSE_OPENCV
 324 endif
 325 ifeq ($(USE_LEVELDB), 1)
 326         COMMON_FLAGS += -DUSE_LEVELDB
 327 endif
 328 ifeq ($(USE_LMDB), 1)
 329         COMMON_FLAGS += -DUSE_LMDB
 330 ifeq ($(ALLOW_LMDB_NOLOCK), 1)
 331         COMMON_FLAGS += -DALLOW_LMDB_NOLOCK
 332 endif
 333 endif
 334
 335 # CPU-only configuration
 336 ifeq ($(CPU_ONLY), 1)
 337         OBJS := $(PROTO_OBJS) $(CXX_OBJS)
 338         TEST_OBJS := $(TEST_CXX_OBJS)
 339         TEST_BINS := $(TEST_CXX_BINS)
 340         ALL_WARNS := $(ALL_CXX_WARNS)
 341         TEST_FILTER := --gtest_filter="-*GPU*"
 342         COMMON_FLAGS += -DCPU_ONLY
 343 endif
 344
 345 # Python layer support
 346 ifeq ($(WITH_PYTHON_LAYER), 1)
 347         COMMON_FLAGS += -DWITH_PYTHON_LAYER
 348         LIBRARIES += $(PYTHON_LIBRARIES)
 349 endif
 350
 351 # BLAS configuration (default = ATLAS)
 352 BLAS ?= atlas
 353 ifeq ($(BLAS), mkl)
 354         # MKL
 355         LIBRARIES += mkl_rt
 356         COMMON_FLAGS += -DUSE_MKL
 357         MKL_DIR ?= /opt/intel/mkl
 358         BLAS_INCLUDE ?= $(MKL_DIR)/include
 359         BLAS_LIB ?= $(MKL_DIR)/lib $(MKL_DIR)/lib/intel64
 360 else ifeq ($(BLAS), open)
 361         # OpenBLAS
 362         LIBRARIES += openblas
 363 else
 364         # ATLAS
 365         ifeq ($(LINUX), 1)
 366                 ifeq ($(BLAS), atlas)
 367                         # Linux simply has cblas and atlas
 368                         LIBRARIES += cblas atlas
 369                 endif
 370         else ifeq ($(OSX), 1)
 371                 # OS X packages atlas as the vecLib framework
 372                 LIBRARIES += cblas
 373                 # 10.10 has accelerate while 10.9 has veclib
 374                 XCODE_CLT_VER := $(shell pkgutil --pkg-info=com.apple.pkg.CLTools_Executables | grep 'version' | sed 's/[^0-9]*\([0-9]\).*/\1/')
 375                 XCODE_CLT_GEQ_6 := $(shell [ $(XCODE_CLT_VER) -gt 5 ] && echo 1)
 376                 ifeq ($(XCODE_CLT_GEQ_6), 1)
 377                         BLAS_INCLUDE ?= /System/Library/Frameworks/Accelerate.framework/Versions/Current/Frameworks/vecLib.framework/Headers/
 378                         LDFLAGS += -framework Accelerate
 379                 else
 380                         BLAS_INCLUDE ?= /System/Library/Frameworks/vecLib.framework/Versions/Current/Headers/
 381                         LDFLAGS += -framework vecLib
 382                 endif
 383         endif
 384 endif
 385 INCLUDE_DIRS += $(BLAS_INCLUDE)
 386 LIBRARY_DIRS += $(BLAS_LIB)
 387
 388 LIBRARY_DIRS += $(LIB_BUILD_DIR)
 389
 390 # Automatic dependency generation (nvcc is handled separately)
 391 CXXFLAGS += -MMD -MP
 392
 393 # Complete build flags.
 394 COMMON_FLAGS += $(foreach includedir,$(INCLUDE_DIRS),-I$(includedir))
 395 CXXFLAGS += -pthread -fPIC $(COMMON_FLAGS) $(WARNINGS)
 396 NVCCFLAGS += -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
 397 # mex may invoke an older gcc that is too liberal with -Wuninitalized
 398 MATLAB_CXXFLAGS := $(CXXFLAGS) -Wno-uninitialized
 399 LINKFLAGS += -pthread -fPIC $(COMMON_FLAGS) $(WARNINGS)
 400
 401 USE_PKG_CONFIG ?= 0
 402 ifeq ($(USE_PKG_CONFIG), 1)
 403         PKG_CONFIG := $(shell pkg-config opencv --libs)
 404 else
 405         PKG_CONFIG :=
 406 endif
 407 LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir)) $(PKG_CONFIG) \
 408                 $(foreach library,$(LIBRARIES),-l$(library))
 409 PYTHON_LDFLAGS := $(LDFLAGS) $(foreach library,$(PYTHON_LIBRARIES),-l$(library))
 410
 411 # 'superclean' target recursively* deletes all files ending with an extension
 412 # in $(SUPERCLEAN_EXTS) below.  This may be useful if you've built older
 413 # versions of Caffe that do not place all generated files in a location known
 414 # to the 'clean' target.
 415 #
 416 # 'supercleanlist' will list the files to be deleted by make superclean.
 417 #
 418 # * Recursive with the exception that symbolic links are never followed, per the
 419 # default behavior of 'find'.
 420 SUPERCLEAN_EXTS := .so .a .o .bin .testbin .pb.cc .pb.h _pb2.py .cuo
 421
 422 # Set the sub-targets of the 'everything' target.
 423 EVERYTHING_TARGETS := all py$(PROJECT) test warn lint
 424 # Only build matcaffe as part of "everything" if MATLAB_DIR is specified.
 425 ifneq ($(MATLAB_DIR),)
 426         EVERYTHING_TARGETS += mat$(PROJECT)
 427 endif
 428
 429 ##############################
 430 # Define build targets
 431 ##############################
 432 .PHONY: all lib test clean docs linecount lint lintclean tools examples $(DIST_ALIASES) \
 433         py mat py$(PROJECT) mat$(PROJECT) proto runtest \
 434         superclean supercleanlist supercleanfiles warn everything
 435
 436 all: lib tools examples
 437
 438 lib: $(STATIC_NAME) $(DYNAMIC_NAME)
 439
 440 everything: $(EVERYTHING_TARGETS)
 441
 442 linecount:
 443         cloc --read-lang-def=$(PROJECT).cloc \
 444                 src/$(PROJECT) include/$(PROJECT) tools examples \
 445                 python matlab
 446
 447 lint: $(EMPTY_LINT_REPORT)
 448
 449 lintclean:
 450         @ $(RM) -r $(LINT_OUTPUT_DIR) $(EMPTY_LINT_REPORT) $(NONEMPTY_LINT_REPORT)
 451
 452 docs: $(DOXYGEN_OUTPUT_DIR)
 453         @ cd ./docs ; ln -sfn ../$(DOXYGEN_OUTPUT_DIR)/html doxygen
 454
 455 $(DOXYGEN_OUTPUT_DIR): $(DOXYGEN_CONFIG_FILE) $(DOXYGEN_SOURCES)
 456         $(DOXYGEN_COMMAND) $(DOXYGEN_CONFIG_FILE)
 457
 458 $(EMPTY_LINT_REPORT): $(LINT_OUTPUTS) | $(BUILD_DIR)
 459         @ cat $(LINT_OUTPUTS) > $@
 460         @ if [ -s "$@" ]; then \
 461                 cat $@; \
 462                 mv $@ $(NONEMPTY_LINT_REPORT); \
 463                 echo "Found one or more lint errors."; \
 464                 exit 1; \
 465           fi; \
 466           $(RM) $(NONEMPTY_LINT_REPORT); \
 467           echo "No lint errors!";
 468
 469 $(LINT_OUTPUTS): $(LINT_OUTPUT_DIR)/%.lint.txt : % $(LINT_SCRIPT) | $(LINT_OUTPUT_DIR)
 470         @ mkdir -p $(dir $@)
 471         @ python $(LINT_SCRIPT) $< 2>&1 \
 472                 | grep -v "^Done processing " \
 473                 | grep -v "^Total errors found: 0" \
 474                 > $@ \
 475                 || true
 476
 477 test: $(TEST_ALL_BIN) $(TEST_ALL_DYNLINK_BIN) $(TEST_BINS)
 478
 479 tools: $(TOOL_BINS) $(TOOL_BIN_LINKS)
 480
 481 examples: $(EXAMPLE_BINS)
 482
 483 py$(PROJECT): py
 484
 485 py: $(PY$(PROJECT)_SO) $(PROTO_GEN_PY)
 486
 487 $(PY$(PROJECT)_SO): $(PY$(PROJECT)_SRC) $(PY$(PROJECT)_HXX) | $(DYNAMIC_NAME)
 488         @ echo CXX/LD -o $@ $<
 489         $(Q)$(CXX) -shared -o $@ $(PY$(PROJECT)_SRC) \
 490                 -o $@ $(LINKFLAGS) -l$(LIBRARY_NAME) $(PYTHON_LDFLAGS) \
 491                 -Wl,-rpath,$(ORIGIN)/../../build/lib
 492
 493 mat$(PROJECT): mat
 494
 495 mat: $(MAT$(PROJECT)_SO)
 496
 497 $(MAT$(PROJECT)_SO): $(MAT$(PROJECT)_SRC) $(STATIC_NAME)
 498         @ if [ -z "$(MATLAB_DIR)" ]; then \
 499                 echo "MATLAB_DIR must be specified in $(CONFIG_FILE)" \
 500                         "to build mat$(PROJECT)."; \
 501                 exit 1; \
 502         fi
 503         @ echo MEX $<
 504         $(Q)$(MATLAB_DIR)/bin/mex $(MAT$(PROJECT)_SRC) \
 505                         CXX="$(CXX)" \
 506                         CXXFLAGS="\$$CXXFLAGS $(MATLAB_CXXFLAGS)" \
 507                         CXXLIBS="\$$CXXLIBS $(STATIC_LINK_COMMAND) $(LDFLAGS)" -output $@
 508         @ if [ -f "$(PROJECT)_.d" ]; then \
 509                 mv -f $(PROJECT)_.d $(BUILD_DIR)/${MAT$(PROJECT)_SO:.$(MAT_SO_EXT)=.d}; \
 510         fi
 511
 512 runtest: $(TEST_ALL_BIN)
 513         $(TOOL_BUILD_DIR)/caffe
 514         $(TEST_ALL_BIN) $(TEST_GPUID) --gtest_shuffle $(TEST_FILTER)
 515
 516 pytest: py
 517         cd python; python -m unittest discover -s caffe/test
 518
 519 mattest: mat
 520         cd matlab; $(MATLAB_DIR)/bin/matlab -nodisplay -r 'caffe.run_tests(), exit()'
 521
 522 warn: $(EMPTY_WARN_REPORT)
 523
 524 $(EMPTY_WARN_REPORT): $(ALL_WARNS) | $(BUILD_DIR)
 525         @ cat $(ALL_WARNS) > $@
 526         @ if [ -s "$@" ]; then \
 527                 cat $@; \
 528                 mv $@ $(NONEMPTY_WARN_REPORT); \
 529                 echo "Compiler produced one or more warnings."; \
 530                 exit 1; \
 531           fi; \
 532           $(RM) $(NONEMPTY_WARN_REPORT); \
 533           echo "No compiler warnings!";
 534
 535 $(ALL_WARNS): %.o.$(WARNS_EXT) : %.o
 536
 537 $(BUILD_DIR_LINK): $(BUILD_DIR)/.linked
 538
 539 # Create a target ".linked" in this BUILD_DIR to tell Make that the "build" link
 540 # is currently correct, then delete the one in the OTHER_BUILD_DIR in case it
 541 # exists and $(DEBUG) is toggled later.
 542 $(BUILD_DIR)/.linked:
 543         @ mkdir -p $(BUILD_DIR)
 544         @ $(RM) $(OTHER_BUILD_DIR)/.linked
 545         @ $(RM) -r $(BUILD_DIR_LINK)
 546         @ ln -s $(BUILD_DIR) $(BUILD_DIR_LINK)
 547         @ touch $@
 548
 549 $(ALL_BUILD_DIRS): | $(BUILD_DIR_LINK)
 550         @ mkdir -p $@
 551
 552 $(DYNAMIC_NAME): $(OBJS) | $(LIB_BUILD_DIR)
 553         @ echo LD -o $@
 554         $(Q)$(CXX) -shared -o $@ $(OBJS) $(VERSIONFLAGS) $(LINKFLAGS) $(LDFLAGS)
 555         @ cd $(BUILD_DIR)/lib; rm -f $(DYNAMIC_NAME_SHORT);   ln -s $(DYNAMIC_VERSIONED_NAME_SHORT) $(DYNAMIC_NAME_SHORT)
 556
 557 $(STATIC_NAME): $(OBJS) | $(LIB_BUILD_DIR)
 558         @ echo AR -o $@
 559         $(Q)ar rcs $@ $(OBJS)
 560
 561 $(BUILD_DIR)/%.o: %.cpp | $(ALL_BUILD_DIRS)
 562         @ echo CXX $<
 563         $(Q)$(CXX) $< $(CXXFLAGS) -c -o $@ 2> $@.$(WARNS_EXT) \
 564                 || (cat $@.$(WARNS_EXT); exit 1)
 565         @ cat $@.$(WARNS_EXT)
 566
 567 $(PROTO_BUILD_DIR)/%.pb.o: $(PROTO_BUILD_DIR)/%.pb.cc $(PROTO_GEN_HEADER) \
 568                 | $(PROTO_BUILD_DIR)
 569         @ echo CXX $<
 570         $(Q)$(CXX) $< $(CXXFLAGS) -c -o $@ 2> $@.$(WARNS_EXT) \
 571                 || (cat $@.$(WARNS_EXT); exit 1)
 572         @ cat $@.$(WARNS_EXT)
 573
 574 $(BUILD_DIR)/cuda/%.o: %.cu | $(ALL_BUILD_DIRS)
 575         @ echo NVCC $<
 576         $(Q)$(CUDA_DIR)/bin/nvcc $(NVCCFLAGS) $(CUDA_ARCH) -M $< -o ${@:.o=.d} \
 577                 -odir $(@D)
 578         $(Q)$(CUDA_DIR)/bin/nvcc $(NVCCFLAGS) $(CUDA_ARCH) -c $< -o $@ 2> $@.$(WARNS_EXT) \
 579                 || (cat $@.$(WARNS_EXT); exit 1)
 580         @ cat $@.$(WARNS_EXT)
 581
 582 $(TEST_ALL_BIN): $(TEST_MAIN_SRC) $(TEST_OBJS) $(GTEST_OBJ) \
 583                 | $(DYNAMIC_NAME) $(TEST_BIN_DIR)
 584         @ echo CXX/LD -o $@ $<
 585         $(Q)$(CXX) $(TEST_MAIN_SRC) $(TEST_OBJS) $(GTEST_OBJ) \
 586                 -o $@ $(LINKFLAGS) $(LDFLAGS) -l$(LIBRARY_NAME) -Wl,-rpath,$(ORIGIN)/../lib
 587
 588 $(TEST_CU_BINS): $(TEST_BIN_DIR)/%.testbin: $(TEST_CU_BUILD_DIR)/%.o \
 589         $(GTEST_OBJ) | $(DYNAMIC_NAME) $(TEST_BIN_DIR)
 590         @ echo LD $<
 591         $(Q)$(CXX) $(TEST_MAIN_SRC) $< $(GTEST_OBJ) \
 592                 -o $@ $(LINKFLAGS) $(LDFLAGS) -l$(LIBRARY_NAME) -Wl,-rpath,$(ORIGIN)/../lib
 593
 594 $(TEST_CXX_BINS): $(TEST_BIN_DIR)/%.testbin: $(TEST_CXX_BUILD_DIR)/%.o \
 595         $(GTEST_OBJ) | $(DYNAMIC_NAME) $(TEST_BIN_DIR)
 596         @ echo LD $<
 597         $(Q)$(CXX) $(TEST_MAIN_SRC) $< $(GTEST_OBJ) \
 598                 -o $@ $(LINKFLAGS) $(LDFLAGS) -l$(LIBRARY_NAME) -Wl,-rpath,$(ORIGIN)/../lib
 599
 600 # Target for extension-less symlinks to tool binaries with extension '*.bin'.
 601 $(TOOL_BUILD_DIR)/%: $(TOOL_BUILD_DIR)/%.bin | $(TOOL_BUILD_DIR)
 602         @ $(RM) $@
 603         @ ln -s $(notdir $<) $@
 604
 605 $(TOOL_BINS): %.bin : %.o | $(DYNAMIC_NAME)
 606         @ echo CXX/LD -o $@
 607         $(Q)$(CXX) $< -o $@ $(LINKFLAGS) -l$(LIBRARY_NAME) $(LDFLAGS) \
 608                 -Wl,-rpath,$(ORIGIN)/../lib
 609
 610 $(EXAMPLE_BINS): %.bin : %.o | $(DYNAMIC_NAME)
 611         @ echo CXX/LD -o $@
 612         $(Q)$(CXX) $< -o $@ $(LINKFLAGS) -l$(LIBRARY_NAME) $(LDFLAGS) \
 613                 -Wl,-rpath,$(ORIGIN)/../../lib
 614
 615 proto: $(PROTO_GEN_CC) $(PROTO_GEN_HEADER)
 616
 617 $(PROTO_BUILD_DIR)/%.pb.cc $(PROTO_BUILD_DIR)/%.pb.h : \
 618                 $(PROTO_SRC_DIR)/%.proto | $(PROTO_BUILD_DIR)
 619         @ echo PROTOC $<
 620         $(Q)protoc --proto_path=$(PROTO_SRC_DIR) --cpp_out=$(PROTO_BUILD_DIR) $<
 621
 622 $(PY_PROTO_BUILD_DIR)/%_pb2.py : $(PROTO_SRC_DIR)/%.proto \
 623                 $(PY_PROTO_INIT) | $(PY_PROTO_BUILD_DIR)
 624         @ echo PROTOC \(python\) $<
 625         $(Q)protoc --proto_path=$(PROTO_SRC_DIR) --python_out=$(PY_PROTO_BUILD_DIR) $<
 626
 627 $(PY_PROTO_INIT): | $(PY_PROTO_BUILD_DIR)
 628         touch $(PY_PROTO_INIT)
 629
 630 clean:
 631         @- $(RM) -rf $(ALL_BUILD_DIRS)
 632         @- $(RM) -rf $(OTHER_BUILD_DIR)
 633         @- $(RM) -rf $(BUILD_DIR_LINK)
 634         @- $(RM) -rf $(DISTRIBUTE_DIR)
 635         @- $(RM) $(PY$(PROJECT)_SO)
 636         @- $(RM) $(MAT$(PROJECT)_SO)
 637
 638 supercleanfiles:
 639         $(eval SUPERCLEAN_FILES := $(strip \
 640                         $(foreach ext,$(SUPERCLEAN_EXTS), $(shell find . -name '*$(ext)' \
 641                         -not -path './data/*'))))
 642
 643 supercleanlist: supercleanfiles
 644         @ \
 645         if [ -z "$(SUPERCLEAN_FILES)" ]; then \
 646                 echo "No generated files found."; \
 647         else \
 648                 echo $(SUPERCLEAN_FILES) | tr ' ' '\n'; \
 649         fi
 650
 651 superclean: clean supercleanfiles
 652         @ \
 653         if [ -z "$(SUPERCLEAN_FILES)" ]; then \
 654                 echo "No generated files found."; \
 655         else \
 656                 echo "Deleting the following generated files:"; \
 657                 echo $(SUPERCLEAN_FILES) | tr ' ' '\n'; \
 658                 $(RM) $(SUPERCLEAN_FILES); \
 659         fi
 660
 661 $(DIST_ALIASES): $(DISTRIBUTE_DIR)
 662
 663 $(DISTRIBUTE_DIR): all py | $(DISTRIBUTE_SUBDIRS)
 664         # add proto
 665         cp -r src/caffe/proto $(DISTRIBUTE_DIR)/
 666         # add include
 667         cp -r include $(DISTRIBUTE_DIR)/
 668         mkdir -p $(DISTRIBUTE_DIR)/include/caffe/proto
 669         cp $(PROTO_GEN_HEADER_SRCS) $(DISTRIBUTE_DIR)/include/caffe/proto
 670         # add tool and example binaries
 671         cp $(TOOL_BINS) $(DISTRIBUTE_DIR)/bin
 672         cp $(EXAMPLE_BINS) $(DISTRIBUTE_DIR)/bin
 673         # add libraries
 674         cp $(STATIC_NAME) $(DISTRIBUTE_DIR)/lib
 675         install -m 644 $(DYNAMIC_NAME) $(DISTRIBUTE_DIR)/lib
 676         cd $(DISTRIBUTE_DIR)/lib; rm -f $(DYNAMIC_NAME_SHORT);   ln -s $(DYNAMIC_VERSIONED_NAME_SHORT) $(DYNAMIC_NAME_SHORT)
 677         # add python - it's not the standard way, indeed...
 678         cp -r python $(DISTRIBUTE_DIR)/python
 679
 680 -include $(DEPS)