From 153705226d8ca746478b69eeac9bc854766daa11 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 27 Jan 2015 21:31:06 +0000
Subject: [PATCH] Bias updates bug fix

---
 Makefile |    5 ++---
 1 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/Makefile b/Makefile
index e48e142..cc0c9ad 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,6 @@
 GPU=1
 DEBUG=0
+ARCH= -arch=sm_35
 
 VPATH=./src/
 EXEC=cnn
@@ -8,7 +9,6 @@
 CC=gcc
 NVCC=nvcc
 OPTS=-O3
-LINKER=$(CC)
 LDFLAGS=`pkg-config --libs opencv` -lm -pthread
 COMMON=`pkg-config --cflags opencv` -I/usr/local/cuda/include/
 CFLAGS=-Wall -Wfatal-errors
@@ -20,7 +20,6 @@
 endif
 
 ifeq ($(GPU), 1) 
-LINKER=$(NVCC)
 COMMON+=-DGPU
 CFLAGS+=-DGPU
 LDFLAGS+= -L/usr/local/cuda/lib64 -lcuda -lcudart -lcublas
@@ -43,7 +42,7 @@
 	$(CC) $(COMMON) $(CFLAGS) -c $< -o $@
 
 $(OBJDIR)%.o: %.cu 
-	$(NVCC) $(COMMON) --compiler-options "$(CFLAGS)" -c $< -o $@
+	$(NVCC) $(ARCH) $(COMMON) --compiler-options "$(CFLAGS)" -c $< -o $@
 
 .PHONY: clean
 

--
Gitblit v1.10.0