From e36182cd8c5dd5c6d0aa1f77cf5cdca87e8bb1f0 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Fri, 21 Nov 2014 23:35:19 +0000 Subject: [PATCH] cleaned up data parsing a lot. probably nothing broken? --- Makefile | 16 ++++++++++++---- 1 files changed, 12 insertions(+), 4 deletions(-) diff --git a/Makefile b/Makefile index b5ad1eb..72ee030 100644 --- a/Makefile +++ b/Makefile @@ -1,12 +1,20 @@ -CC=gcc GPU=1 +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 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 ifeq ($(GPU), 1) @@ -15,7 +23,7 @@ else OPTS+= -march=native ifeq ($(GPU), 1) -LDFLAGS= -lOpenCL +LDFLAGS+= -lOpenCL endif endif CFLAGS= $(COMMON) $(OPTS) @@ -25,7 +33,7 @@ EXEC=cnn OBJDIR=./obj/ -OBJ=network.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 +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 OBJS = $(addprefix $(OBJDIR), $(OBJ)) all: $(EXEC) -- Gitblit v1.10.0