From aa5996d58e68edfbefe51061856aecd549dd09c4 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Tue, 13 Jan 2015 01:27:08 +0000 Subject: [PATCH] Faster --- Makefile | 31 +++++++++++++++++++++---------- 1 files changed, 21 insertions(+), 10 deletions(-) diff --git a/Makefile b/Makefile index 445c775..3bf04a3 100644 --- a/Makefile +++ b/Makefile @@ -1,11 +1,19 @@ -CC=gcc GPU=1 -COMMON=-Wall -Werror -Wfatal-errors `pkg-config --cflags opencv` -I/usr/local/cuda/include/ +CLBLAS=0 + +CC=gcc +COMMON=-Wall -Wfatal-errors `pkg-config --cflags opencv` -I/usr/local/cuda/include/ ifeq ($(GPU), 1) COMMON+=-DGPU -else endif + +ifeq ($(CLBLAS), 1) +COMMON+=-DCLBLAS +LDFLAGS=-lclBLAS +endif + UNAME = $(shell uname) +#OPTS=-Ofast -flto OPTS=-O3 -flto ifeq ($(UNAME), Darwin) COMMON+= -isystem /usr/local/Cellar/opencv/2.4.6.1/include/opencv -isystem /usr/local/Cellar/opencv/2.4.6.1/include @@ -13,28 +21,31 @@ LDFLAGS= -framework OpenCL endif else +OPTS+= -march=native ifeq ($(GPU), 1) -LDFLAGS= -lOpenCL +LDFLAGS+= -lOpenCL endif endif CFLAGS= $(COMMON) $(OPTS) -#CFLAGS= $(COMMON) -O0 -g -LDFLAGS+=`pkg-config --libs opencv` -lm +#CFLAGS= $(COMMON) -O0 -g +LDFLAGS+=`pkg-config --libs opencv` -lm -pthread VPATH=./src/ EXEC=cnn +OBJDIR=./obj/ -OBJ=network.o image.o tests.o connected_layer.o maxpool_layer.o activations.o list.o option_list.o parser.o utils.o data.o matrix.o softmax_layer.o mini_blas.o convolutional_layer.o gemm.o normalization_layer.o opencl.o im2col.o col2im.o axpy.o +OBJ=network.o network_gpu.o image.o cnn.o connected_layer.o maxpool_layer.o activations.o list.o option_list.o parser.o utils.o data.o matrix.o softmax_layer.o mini_blas.o convolutional_layer.o gemm.o normalization_layer.o opencl.o im2col.o col2im.o axpy.o dropout_layer.o crop_layer.o freeweight_layer.o cost_layer.o server.o +OBJS = $(addprefix $(OBJDIR), $(OBJ)) all: $(EXEC) -$(EXEC): $(OBJ) +$(EXEC): $(OBJS) $(CC) $(CFLAGS) $(LDFLAGS) $^ -o $@ -%.o: %.c +$(OBJDIR)%.o: %.c $(CC) $(CFLAGS) -c $< -o $@ .PHONY: clean clean: - rm -rf $(OBJ) $(EXEC) + rm -rf $(OBJS) $(EXEC) -- Gitblit v1.10.0