From 6e1d5b45de988bb795c4c505f22f2170a78b7746 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 20 Jan 2015 06:06:18 +0000
Subject: [PATCH] fast sort of working
---
src/convolutional_layer.cl | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/convolutional_layer.cl b/src/convolutional_layer.cl
index 3b091cf..903471b 100644
--- a/src/convolutional_layer.cl
+++ b/src/convolutional_layer.cl
@@ -19,7 +19,7 @@
for(b = 0; b < batch; ++b){
for(i = 0; i < size; i += BLOCK){
int index = p + i + size*(filter + n*b);
- sum += (index < size) ? delta[index] : 0;
+ sum += (p+i < size) ? delta[index] : 0;
}
}
part[p] = sum;
--
Gitblit v1.10.0