| CFLAGS += -I$(CAFFE_PATH)/include -I$(CAFFE_PATH)/build/src | |
| LDFLAGS += -lprotobuf -lboost_system -lboost_thread -lboost_filesystem -lgflags -lglog -L$(CAFFE_PATH)/build/lib -lcaffe | |
| ifeq ($(USE_CUDNN), 1) | |
| CFLAGS += -DUSE_CUDNN=1 | |
| endif | |
| ifeq ($(USE_CUDA), 0) | |
| CFLAGS += -DCPU_ONLY=1 | |
| endif | |
| CAFFE_SRC = $(wildcard plugin/caffe/*.cc) | |
| PLUGIN_OBJ += $(patsubst %.cc, build/%.o, $(CAFFE_SRC)) | |
| CAFFE_CUSRC = $(wildcard plugin/caffe/*.cu) | |
| PLUGIN_CUOBJ += $(patsubst %.cu, build/%_gpu.o, $(CAFFE_CUSRC)) |