From 9e98f1b68e2ebf618d7f65a0985f5efd11e413f9 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Thu, 18 Dec 2014 18:47:33 +0000 Subject: [PATCH] Not sure --- Makefile | 18 +++++++++++++----- 1 files changed, 13 insertions(+), 5 deletions(-) diff --git a/Makefile b/Makefile index b5ad1eb..3247999 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,17 +23,17 @@ 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 +LDFLAGS+=`pkg-config --libs opencv` -lm -pthread VPATH=./src/ 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 server.o OBJS = $(addprefix $(OBJDIR), $(OBJ)) all: $(EXEC) -- Gitblit v1.10.0