caffe

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
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
182

183
# handle IO dependencies
184
USE_LEVELDB ?= 1
185
USE_LMDB ?= 1
186
# This code is taken from https://github.com/sh1r0/caffe-android-lib
187
USE_HDF5 ?= 1
188
USE_OPENCV ?= 1
189

190
ifeq ($(USE_LEVELDB), 1)
191
	LIBRARIES += leveldb snappy
192
endif
193
ifeq ($(USE_LMDB), 1)
194
	LIBRARIES += lmdb
195
endif
196
# This code is taken from https://github.com/sh1r0/caffe-android-lib
197
ifeq ($(USE_HDF5), 1)
198
	LIBRARIES += hdf5_hl hdf5
199
endif
200
ifeq ($(USE_OPENCV), 1)
201
	LIBRARIES += opencv_core opencv_highgui opencv_imgproc
202

203
	ifeq ($(OPENCV_VERSION), 3)
204
		LIBRARIES += opencv_imgcodecs
205
	endif
206

207
endif
208
PYTHON_LIBRARIES ?= boost_python python2.7
209
WARNINGS := -Wall -Wno-sign-compare
210

211
##############################
212
# Set build directories
213
##############################
214

215
DISTRIBUTE_DIR ?= distribute
216
DISTRIBUTE_SUBDIRS := $(DISTRIBUTE_DIR)/bin $(DISTRIBUTE_DIR)/lib
217
DIST_ALIASES := dist
218
ifneq ($(strip $(DISTRIBUTE_DIR)),distribute)
219
		DIST_ALIASES += distribute
220
endif
221

222
ALL_BUILD_DIRS := $(sort $(BUILD_DIR) $(addprefix $(BUILD_DIR)/, $(SRC_DIRS)) \
223
	$(addprefix $(BUILD_DIR)/cuda/, $(SRC_DIRS)) \
224
	$(LIB_BUILD_DIR) $(TEST_BIN_DIR) $(PY_PROTO_BUILD_DIR) $(LINT_OUTPUT_DIR) \
225
	$(DISTRIBUTE_SUBDIRS) $(PROTO_BUILD_INCLUDE_DIR))
226

227
##############################
228
# Set directory for Doxygen-generated documentation
229
##############################
230
DOXYGEN_CONFIG_FILE ?= ./.Doxyfile
231
# should be the same as OUTPUT_DIRECTORY in the .Doxyfile
232
DOXYGEN_OUTPUT_DIR ?= ./doxygen
233
DOXYGEN_COMMAND ?= doxygen
234
# All the files that might have Doxygen documentation.
235
DOXYGEN_SOURCES := $(shell find \
236
	src/$(PROJECT) \
237
	include/$(PROJECT) \
238
	python/ \
239
	matlab/ \
240
	examples \
241
	tools \
242
	-name "*.cpp" -or -name "*.hpp" -or -name "*.cu" -or -name "*.cuh" -or \
243
        -name "*.py" -or -name "*.m")
244
DOXYGEN_SOURCES += $(DOXYGEN_CONFIG_FILE)
245

246

247
##############################
248
# Configure build
249
##############################
250

251
# Determine platform
252
UNAME := $(shell uname -s)
253
ifeq ($(UNAME), Linux)
254
	LINUX := 1
255
else ifeq ($(UNAME), Darwin)
256
	OSX := 1
257
	OSX_MAJOR_VERSION := $(shell sw_vers -productVersion | cut -f 1 -d .)
258
	OSX_MINOR_VERSION := $(shell sw_vers -productVersion | cut -f 2 -d .)
259
endif
260

261
# Linux
262
ifeq ($(LINUX), 1)
263
	CXX ?= /usr/bin/g++
264
	GCCVERSION := $(shell $(CXX) -dumpversion | cut -f1,2 -d.)
265
	# older versions of gcc are too dumb to build boost with -Wuninitalized
266
	ifeq ($(shell echo | awk '{exit $(GCCVERSION) < 4.6;}'), 1)
267
		WARNINGS += -Wno-uninitialized
268
	endif
269
	# boost::thread is reasonably called boost_thread (compare OS X)
270
	# We will also explicitly add stdc++ to the link target.
271
	LIBRARIES += boost_thread stdc++
272
	VERSIONFLAGS += -Wl,-soname,$(DYNAMIC_VERSIONED_NAME_SHORT) -Wl,-rpath,$(ORIGIN)/../lib
273
endif
274

275
# OS X:
276
# clang++ instead of g++
277
# libstdc++ for NVCC compatibility on OS X >= 10.9 with CUDA < 7.0
278
ifeq ($(OSX), 1)
279
	CXX := /usr/bin/clang++
280
	ifneq ($(CPU_ONLY), 1)
281
		CUDA_VERSION := $(shell $(CUDA_DIR)/bin/nvcc -V | grep -o 'release [0-9.]*' | tr -d '[a-z ]')
282
		ifeq ($(shell echo | awk '{exit $(CUDA_VERSION) < 7.0;}'), 1)
283
			CXXFLAGS += -stdlib=libstdc++
284
			LINKFLAGS += -stdlib=libstdc++
285
		endif
286
		# clang throws this warning for cuda headers
287
		WARNINGS += -Wno-unneeded-internal-declaration
288
		# 10.11 strips DYLD_* env vars so link CUDA (rpath is available on 10.5+)
289
		OSX_10_OR_LATER   := $(shell [ $(OSX_MAJOR_VERSION) -ge 10 ] && echo true)
290
		OSX_10_5_OR_LATER := $(shell [ $(OSX_MINOR_VERSION) -ge 5 ] && echo true)
291
		ifeq ($(OSX_10_OR_LATER),true)
292
			ifeq ($(OSX_10_5_OR_LATER),true)
293
				LDFLAGS += -Wl,-rpath,$(CUDA_LIB_DIR)
294
			endif
295
		endif
296
	endif
297
	# gtest needs to use its own tuple to not conflict with clang
298
	COMMON_FLAGS += -DGTEST_USE_OWN_TR1_TUPLE=1
299
	# boost::thread is called boost_thread-mt to mark multithreading on OS X
300
	LIBRARIES += boost_thread-mt
301
	# we need to explicitly ask for the rpath to be obeyed
302
	ORIGIN := @loader_path
303
	VERSIONFLAGS += -Wl,-install_name,@rpath/$(DYNAMIC_VERSIONED_NAME_SHORT) -Wl,-rpath,$(ORIGIN)/../../build/lib
304
else
305
	ORIGIN := \$$ORIGIN
306
endif
307

308
# Custom compiler
309
ifdef CUSTOM_CXX
310
	CXX := $(CUSTOM_CXX)
311
endif
312

313
# Static linking
314
ifneq (,$(findstring clang++,$(CXX)))
315
	STATIC_LINK_COMMAND := -Wl,-force_load $(STATIC_NAME)
316
else ifneq (,$(findstring g++,$(CXX)))
317
	STATIC_LINK_COMMAND := -Wl,--whole-archive $(STATIC_NAME) -Wl,--no-whole-archive
318
else
319
  # The following line must not be indented with a tab, since we are not inside a target
320
  $(error Cannot static link with the $(CXX) compiler)
321
endif
322

323
# Debugging
324
ifeq ($(DEBUG), 1)
325
	COMMON_FLAGS += -DDEBUG -g -O0
326
	NVCCFLAGS += -G
327
else
328
	COMMON_FLAGS += -DNDEBUG -O2
329
endif
330

331
# cuDNN acceleration configuration.
332
ifeq ($(USE_CUDNN), 1)
333
	LIBRARIES += cudnn
334
	COMMON_FLAGS += -DUSE_CUDNN
335
endif
336

337
# NCCL acceleration configuration
338
ifeq ($(USE_NCCL), 1)
339
	LIBRARIES += nccl
340
	COMMON_FLAGS += -DUSE_NCCL
341
endif
342

343
# configure IO libraries
344
ifeq ($(USE_OPENCV), 1)
345
	COMMON_FLAGS += -DUSE_OPENCV
346
endif
347
ifeq ($(USE_LEVELDB), 1)
348
	COMMON_FLAGS += -DUSE_LEVELDB
349
endif
350
ifeq ($(USE_LMDB), 1)
351
	COMMON_FLAGS += -DUSE_LMDB
352
ifeq ($(ALLOW_LMDB_NOLOCK), 1)
353
	COMMON_FLAGS += -DALLOW_LMDB_NOLOCK
354
endif
355
endif
356
# This code is taken from https://github.com/sh1r0/caffe-android-lib
357
ifeq ($(USE_HDF5), 1)
358
	COMMON_FLAGS += -DUSE_HDF5
359
endif
360

361
# CPU-only configuration
362
ifeq ($(CPU_ONLY), 1)
363
	OBJS := $(PROTO_OBJS) $(CXX_OBJS)
364
	TEST_OBJS := $(TEST_CXX_OBJS)
365
	TEST_BINS := $(TEST_CXX_BINS)
366
	ALL_WARNS := $(ALL_CXX_WARNS)
367
	TEST_FILTER := --gtest_filter="-*GPU*"
368
	COMMON_FLAGS += -DCPU_ONLY
369
endif
370

371
# Python layer support
372
ifeq ($(WITH_PYTHON_LAYER), 1)
373
	COMMON_FLAGS += -DWITH_PYTHON_LAYER
374
	LIBRARIES += $(PYTHON_LIBRARIES)
375
endif
376

377
# BLAS configuration (default = ATLAS)
378
BLAS ?= atlas
379
ifeq ($(BLAS), mkl)
380
	# MKL
381
	LIBRARIES += mkl_rt
382
	COMMON_FLAGS += -DUSE_MKL
383
	MKLROOT ?= /opt/intel/mkl
384
	BLAS_INCLUDE ?= $(MKLROOT)/include
385
	BLAS_LIB ?= $(MKLROOT)/lib $(MKLROOT)/lib/intel64
386
else ifeq ($(BLAS), open)
387
	# OpenBLAS
388
	LIBRARIES += openblas
389
else
390
	# ATLAS
391
	ifeq ($(LINUX), 1)
392
		ifeq ($(BLAS), atlas)
393
			# Linux simply has cblas and atlas
394
			LIBRARIES += cblas atlas
395
		endif
396
	else ifeq ($(OSX), 1)
397
		# OS X packages atlas as the vecLib framework
398
		LIBRARIES += cblas
399
		# 10.10 has accelerate while 10.9 has veclib
400
		XCODE_CLT_VER := $(shell pkgutil --pkg-info=com.apple.pkg.CLTools_Executables | grep 'version' | sed 's/[^0-9]*\([0-9]\).*/\1/')
401
		XCODE_CLT_GEQ_7 := $(shell [ $(XCODE_CLT_VER) -gt 6 ] && echo 1)
402
		XCODE_CLT_GEQ_6 := $(shell [ $(XCODE_CLT_VER) -gt 5 ] && echo 1)
403
		ifeq ($(XCODE_CLT_GEQ_7), 1)
404
			BLAS_INCLUDE ?= /Applications/Xcode.app/Contents/Developer/Platforms/MacOSX.platform/Developer/SDKs/$(shell ls /Applications/Xcode.app/Contents/Developer/Platforms/MacOSX.platform/Developer/SDKs/ | sort | tail -1)/System/Library/Frameworks/Accelerate.framework/Versions/A/Frameworks/vecLib.framework/Versions/A/Headers
405
		else ifeq ($(XCODE_CLT_GEQ_6), 1)
406
			BLAS_INCLUDE ?= /System/Library/Frameworks/Accelerate.framework/Versions/Current/Frameworks/vecLib.framework/Headers/
407
			LDFLAGS += -framework Accelerate
408
		else
409
			BLAS_INCLUDE ?= /System/Library/Frameworks/vecLib.framework/Versions/Current/Headers/
410
			LDFLAGS += -framework vecLib
411
		endif
412
	endif
413
endif
414
INCLUDE_DIRS += $(BLAS_INCLUDE)
415
LIBRARY_DIRS += $(BLAS_LIB)
416

417
LIBRARY_DIRS += $(LIB_BUILD_DIR)
418

419
# Automatic dependency generation (nvcc is handled separately)
420
CXXFLAGS += -MMD -MP
421

422
# Complete build flags.
423
COMMON_FLAGS += $(foreach includedir,$(INCLUDE_DIRS),-I$(includedir))
424
CXXFLAGS += -pthread -fPIC $(COMMON_FLAGS) $(WARNINGS)
425
NVCCFLAGS += -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
426
# mex may invoke an older gcc that is too liberal with -Wuninitalized
427
MATLAB_CXXFLAGS := $(CXXFLAGS) -Wno-uninitialized
428
LINKFLAGS += -pthread -fPIC $(COMMON_FLAGS) $(WARNINGS)
429

430
USE_PKG_CONFIG ?= 0
431
ifeq ($(USE_PKG_CONFIG), 1)
432
	PKG_CONFIG := $(shell pkg-config opencv --libs)
433
else
434
	PKG_CONFIG :=
435
endif
436
LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir)) $(PKG_CONFIG) \
437
		$(foreach library,$(LIBRARIES),-l$(library))
438
PYTHON_LDFLAGS := $(LDFLAGS) $(foreach library,$(PYTHON_LIBRARIES),-l$(library))
439

440
# 'superclean' target recursively* deletes all files ending with an extension
441
# in $(SUPERCLEAN_EXTS) below.  This may be useful if you've built older
442
# versions of Caffe that do not place all generated files in a location known
443
# to the 'clean' target.
444
#
445
# 'supercleanlist' will list the files to be deleted by make superclean.
446
#
447
# * Recursive with the exception that symbolic links are never followed, per the
448
# default behavior of 'find'.
449
SUPERCLEAN_EXTS := .so .a .o .bin .testbin .pb.cc .pb.h _pb2.py .cuo
450

451
# Set the sub-targets of the 'everything' target.
452
EVERYTHING_TARGETS := all py$(PROJECT) test warn lint
453
# Only build matcaffe as part of "everything" if MATLAB_DIR is specified.
454
ifneq ($(MATLAB_DIR),)
455
	EVERYTHING_TARGETS += mat$(PROJECT)
456
endif
457

458
##############################
459
# Define build targets
460
##############################
461
.PHONY: all lib test clean docs linecount lint lintclean tools examples $(DIST_ALIASES) \
462
	py mat py$(PROJECT) mat$(PROJECT) proto runtest \
463
	superclean supercleanlist supercleanfiles warn everything
464

465
all: lib tools examples
466

467
lib: $(STATIC_NAME) $(DYNAMIC_NAME)
468

469
everything: $(EVERYTHING_TARGETS)
470

471
linecount:
472
	cloc --read-lang-def=$(PROJECT).cloc \
473
		src/$(PROJECT) include/$(PROJECT) tools examples \
474
		python matlab
475

476
lint: $(EMPTY_LINT_REPORT)
477

478
lintclean:
479
	@ $(RM) -r $(LINT_OUTPUT_DIR) $(EMPTY_LINT_REPORT) $(NONEMPTY_LINT_REPORT)
480

481
docs: $(DOXYGEN_OUTPUT_DIR)
482
	@ cd ./docs ; ln -sfn ../$(DOXYGEN_OUTPUT_DIR)/html doxygen
483

484
$(DOXYGEN_OUTPUT_DIR): $(DOXYGEN_CONFIG_FILE) $(DOXYGEN_SOURCES)
485
	$(DOXYGEN_COMMAND) $(DOXYGEN_CONFIG_FILE)
486

487
$(EMPTY_LINT_REPORT): $(LINT_OUTPUTS) | $(BUILD_DIR)
488
	@ cat $(LINT_OUTPUTS) > $@
489
	@ if [ -s "$@" ]; then \
490
		cat $@; \
491
		mv $@ $(NONEMPTY_LINT_REPORT); \
492
		echo "Found one or more lint errors."; \
493
		exit 1; \
494
	  fi; \
495
	  $(RM) $(NONEMPTY_LINT_REPORT); \
496
	  echo "No lint errors!";
497

498
$(LINT_OUTPUTS): $(LINT_OUTPUT_DIR)/%.lint.txt : % $(LINT_SCRIPT) | $(LINT_OUTPUT_DIR)
499
	@ mkdir -p $(dir $@)
500
	@ python $(LINT_SCRIPT) $< 2>&1 \
501
		| grep -v "^Done processing " \
502
		| grep -v "^Total errors found: 0" \
503
		> $@ \
504
		|| true
505

506
test: $(TEST_ALL_BIN) $(TEST_ALL_DYNLINK_BIN) $(TEST_BINS)
507

508
tools: $(TOOL_BINS) $(TOOL_BIN_LINKS)
509

510
examples: $(EXAMPLE_BINS)
511

512
py$(PROJECT): py
513

514
py: $(PY$(PROJECT)_SO) $(PROTO_GEN_PY)
515

516
$(PY$(PROJECT)_SO): $(PY$(PROJECT)_SRC) $(PY$(PROJECT)_HXX) | $(DYNAMIC_NAME)
517
	@ echo CXX/LD -o $@ $<
518
	$(Q)$(CXX) -shared -o $@ $(PY$(PROJECT)_SRC) \
519
		-o $@ $(LINKFLAGS) -l$(LIBRARY_NAME) $(PYTHON_LDFLAGS) \
520
		-Wl,-rpath,$(ORIGIN)/../../build/lib
521

522
mat$(PROJECT): mat
523

524
mat: $(MAT$(PROJECT)_SO)
525

526
$(MAT$(PROJECT)_SO): $(MAT$(PROJECT)_SRC) $(STATIC_NAME)
527
	@ if [ -z "$(MATLAB_DIR)" ]; then \
528
		echo "MATLAB_DIR must be specified in $(CONFIG_FILE)" \
529
			"to build mat$(PROJECT)."; \
530
		exit 1; \
531
	fi
532
	@ echo MEX $<
533
	$(Q)$(MATLAB_DIR)/bin/mex $(MAT$(PROJECT)_SRC) \
534
			CXX="$(CXX)" \
535
			CXXFLAGS="\$$CXXFLAGS $(MATLAB_CXXFLAGS)" \
536
			CXXLIBS="\$$CXXLIBS $(STATIC_LINK_COMMAND) $(LDFLAGS)" -output $@
537
	@ if [ -f "$(PROJECT)_.d" ]; then \
538
		mv -f $(PROJECT)_.d $(BUILD_DIR)/${MAT$(PROJECT)_SO:.$(MAT_SO_EXT)=.d}; \
539
	fi
540

541
runtest: $(TEST_ALL_BIN)
542
	$(TOOL_BUILD_DIR)/caffe
543
	$(TEST_ALL_BIN) $(TEST_GPUID) --gtest_shuffle $(TEST_FILTER)
544

545
pytest: py
546
	cd python; python -m unittest discover -s caffe/test
547

548
mattest: mat
549
	cd matlab; $(MATLAB_DIR)/bin/matlab -nodisplay -r 'caffe.run_tests(), exit()'
550

551
warn: $(EMPTY_WARN_REPORT)
552

553
$(EMPTY_WARN_REPORT): $(ALL_WARNS) | $(BUILD_DIR)
554
	@ cat $(ALL_WARNS) > $@
555
	@ if [ -s "$@" ]; then \
556
		cat $@; \
557
		mv $@ $(NONEMPTY_WARN_REPORT); \
558
		echo "Compiler produced one or more warnings."; \
559
		exit 1; \
560
	  fi; \
561
	  $(RM) $(NONEMPTY_WARN_REPORT); \
562
	  echo "No compiler warnings!";
563

564
$(ALL_WARNS): %.o.$(WARNS_EXT) : %.o
565

566
$(BUILD_DIR_LINK): $(BUILD_DIR)/.linked
567

568
# Create a target ".linked" in this BUILD_DIR to tell Make that the "build" link
569
# is currently correct, then delete the one in the OTHER_BUILD_DIR in case it
570
# exists and $(DEBUG) is toggled later.
571
$(BUILD_DIR)/.linked:
572
	@ mkdir -p $(BUILD_DIR)
573
	@ $(RM) $(OTHER_BUILD_DIR)/.linked
574
	@ $(RM) -r $(BUILD_DIR_LINK)
575
	@ ln -s $(BUILD_DIR) $(BUILD_DIR_LINK)
576
	@ touch $@
577

578
$(ALL_BUILD_DIRS): | $(BUILD_DIR_LINK)
579
	@ mkdir -p $@
580

581
$(DYNAMIC_NAME): $(OBJS) | $(LIB_BUILD_DIR)
582
	@ echo LD -o $@
583
	$(Q)$(CXX) -shared -o $@ $(OBJS) $(VERSIONFLAGS) $(LINKFLAGS) $(LDFLAGS)
584
	@ cd $(BUILD_DIR)/lib; rm -f $(DYNAMIC_NAME_SHORT);   ln -s $(DYNAMIC_VERSIONED_NAME_SHORT) $(DYNAMIC_NAME_SHORT)
585

586
$(STATIC_NAME): $(OBJS) | $(LIB_BUILD_DIR)
587
	@ echo AR -o $@
588
	$(Q)ar rcs $@ $(OBJS)
589

590
$(BUILD_DIR)/%.o: %.cpp $(PROTO_GEN_HEADER) | $(ALL_BUILD_DIRS)
591
	@ echo CXX $<
592
	$(Q)$(CXX) $< $(CXXFLAGS) -c -o $@ 2> $@.$(WARNS_EXT) \
593
		|| (cat $@.$(WARNS_EXT); exit 1)
594
	@ cat $@.$(WARNS_EXT)
595

596
$(PROTO_BUILD_DIR)/%.pb.o: $(PROTO_BUILD_DIR)/%.pb.cc $(PROTO_GEN_HEADER) \
597
		| $(PROTO_BUILD_DIR)
598
	@ echo CXX $<
599
	$(Q)$(CXX) $< $(CXXFLAGS) -c -o $@ 2> $@.$(WARNS_EXT) \
600
		|| (cat $@.$(WARNS_EXT); exit 1)
601
	@ cat $@.$(WARNS_EXT)
602

603
$(BUILD_DIR)/cuda/%.o: %.cu | $(ALL_BUILD_DIRS)
604
	@ echo NVCC $<
605
	$(Q)$(CUDA_DIR)/bin/nvcc $(NVCCFLAGS) $(CUDA_ARCH) -M $< -o ${@:.o=.d} \
606
		-odir $(@D)
607
	$(Q)$(CUDA_DIR)/bin/nvcc $(NVCCFLAGS) $(CUDA_ARCH) -c $< -o $@ 2> $@.$(WARNS_EXT) \
608
		|| (cat $@.$(WARNS_EXT); exit 1)
609
	@ cat $@.$(WARNS_EXT)
610

611
$(TEST_ALL_BIN): $(TEST_MAIN_SRC) $(TEST_OBJS) $(GTEST_OBJ) \
612
		| $(DYNAMIC_NAME) $(TEST_BIN_DIR)
613
	@ echo CXX/LD -o $@ $<
614
	$(Q)$(CXX) $(TEST_MAIN_SRC) $(TEST_OBJS) $(GTEST_OBJ) \
615
		-o $@ $(LINKFLAGS) $(LDFLAGS) -l$(LIBRARY_NAME) -Wl,-rpath,$(ORIGIN)/../lib
616

617
$(TEST_CU_BINS): $(TEST_BIN_DIR)/%.testbin: $(TEST_CU_BUILD_DIR)/%.o \
618
	$(GTEST_OBJ) | $(DYNAMIC_NAME) $(TEST_BIN_DIR)
619
	@ echo LD $<
620
	$(Q)$(CXX) $(TEST_MAIN_SRC) $< $(GTEST_OBJ) \
621
		-o $@ $(LINKFLAGS) $(LDFLAGS) -l$(LIBRARY_NAME) -Wl,-rpath,$(ORIGIN)/../lib
622

623
$(TEST_CXX_BINS): $(TEST_BIN_DIR)/%.testbin: $(TEST_CXX_BUILD_DIR)/%.o \
624
	$(GTEST_OBJ) | $(DYNAMIC_NAME) $(TEST_BIN_DIR)
625
	@ echo LD $<
626
	$(Q)$(CXX) $(TEST_MAIN_SRC) $< $(GTEST_OBJ) \
627
		-o $@ $(LINKFLAGS) $(LDFLAGS) -l$(LIBRARY_NAME) -Wl,-rpath,$(ORIGIN)/../lib
628

629
# Target for extension-less symlinks to tool binaries with extension '*.bin'.
630
$(TOOL_BUILD_DIR)/%: $(TOOL_BUILD_DIR)/%.bin | $(TOOL_BUILD_DIR)
631
	@ $(RM) $@
632
	@ ln -s $(notdir $<) $@
633

634
$(TOOL_BINS): %.bin : %.o | $(DYNAMIC_NAME)
635
	@ echo CXX/LD -o $@
636
	$(Q)$(CXX) $< -o $@ $(LINKFLAGS) -l$(LIBRARY_NAME) $(LDFLAGS) \
637
		-Wl,-rpath,$(ORIGIN)/../lib
638

639
$(EXAMPLE_BINS): %.bin : %.o | $(DYNAMIC_NAME)
640
	@ echo CXX/LD -o $@
641
	$(Q)$(CXX) $< -o $@ $(LINKFLAGS) -l$(LIBRARY_NAME) $(LDFLAGS) \
642
		-Wl,-rpath,$(ORIGIN)/../../lib
643

644
proto: $(PROTO_GEN_CC) $(PROTO_GEN_HEADER)
645

646
$(PROTO_BUILD_DIR)/%.pb.cc $(PROTO_BUILD_DIR)/%.pb.h : \
647
		$(PROTO_SRC_DIR)/%.proto | $(PROTO_BUILD_DIR)
648
	@ echo PROTOC $<
649
	$(Q)protoc --proto_path=$(PROTO_SRC_DIR) --cpp_out=$(PROTO_BUILD_DIR) $<
650

651
$(PY_PROTO_BUILD_DIR)/%_pb2.py : $(PROTO_SRC_DIR)/%.proto \
652
		$(PY_PROTO_INIT) | $(PY_PROTO_BUILD_DIR)
653
	@ echo PROTOC \(python\) $<
654
	$(Q)protoc --proto_path=src --python_out=python $<
655

656
$(PY_PROTO_INIT): | $(PY_PROTO_BUILD_DIR)
657
	touch $(PY_PROTO_INIT)
658

659
clean:
660
	@- $(RM) -rf $(ALL_BUILD_DIRS)
661
	@- $(RM) -rf $(OTHER_BUILD_DIR)
662
	@- $(RM) -rf $(BUILD_DIR_LINK)
663
	@- $(RM) -rf $(DISTRIBUTE_DIR)
664
	@- $(RM) $(PY$(PROJECT)_SO)
665
	@- $(RM) $(MAT$(PROJECT)_SO)
666

667
supercleanfiles:
668
	$(eval SUPERCLEAN_FILES := $(strip \
669
			$(foreach ext,$(SUPERCLEAN_EXTS), $(shell find . -name '*$(ext)' \
670
			-not -path './data/*'))))
671

672
supercleanlist: supercleanfiles
673
	@ \
674
	if [ -z "$(SUPERCLEAN_FILES)" ]; then \
675
		echo "No generated files found."; \
676
	else \
677
		echo $(SUPERCLEAN_FILES) | tr ' ' '\n'; \
678
	fi
679

680
superclean: clean supercleanfiles
681
	@ \
682
	if [ -z "$(SUPERCLEAN_FILES)" ]; then \
683
		echo "No generated files found."; \
684
	else \
685
		echo "Deleting the following generated files:"; \
686
		echo $(SUPERCLEAN_FILES) | tr ' ' '\n'; \
687
		$(RM) $(SUPERCLEAN_FILES); \
688
	fi
689

690
$(DIST_ALIASES): $(DISTRIBUTE_DIR)
691

692
$(DISTRIBUTE_DIR): all py | $(DISTRIBUTE_SUBDIRS)
693
	# add proto
694
	cp -r src/caffe/proto $(DISTRIBUTE_DIR)/
695
	# add include
696
	cp -r include $(DISTRIBUTE_DIR)/
697
	mkdir -p $(DISTRIBUTE_DIR)/include/caffe/proto
698
	cp $(PROTO_GEN_HEADER_SRCS) $(DISTRIBUTE_DIR)/include/caffe/proto
699
	# add tool and example binaries
700
	cp $(TOOL_BINS) $(DISTRIBUTE_DIR)/bin
701
	cp $(EXAMPLE_BINS) $(DISTRIBUTE_DIR)/bin
702
	# add libraries
703
	cp $(STATIC_NAME) $(DISTRIBUTE_DIR)/lib
704
	install -m 644 $(DYNAMIC_NAME) $(DISTRIBUTE_DIR)/lib
705
	cd $(DISTRIBUTE_DIR)/lib; rm -f $(DYNAMIC_NAME_SHORT);   ln -s $(DYNAMIC_VERSIONED_NAME_SHORT) $(DYNAMIC_NAME_SHORT)
706
	# add python - it's not the standard way, indeed...
707
	cp -r python $(DISTRIBUTE_DIR)/
708

709
-include $(DEPS)
710