| CFLAGS += -I$(CAFFE_PATH)/include -I$(CAFFE_PATH)/build/src -I$(CAFFE_PATH)/build/include |
| 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)) |
