From d00f0a1ccd2a9b1c332bbf7754f291dd61dee14f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 21 Jul 2015 23:09:33 +0000
Subject: [PATCH] Changes to make routing work better
---
src/route_layer.c | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/route_layer.c b/src/route_layer.c
index e3802b7..67b606c 100644
--- a/src/route_layer.c
+++ b/src/route_layer.c
@@ -54,7 +54,7 @@
float *delta = net.layers[index].delta;
int input_size = l.input_sizes[i];
for(j = 0; j < l.batch; ++j){
- copy_cpu(input_size, l.delta + offset + j*l.outputs, 1, delta + j*input_size, 1);
+ axpy_cpu(input_size, 1, l.delta + offset + j*l.outputs, 1, delta + j*input_size, 1);
}
offset += input_size;
}
@@ -85,7 +85,7 @@
float *delta = net.layers[index].delta_gpu;
int input_size = l.input_sizes[i];
for(j = 0; j < l.batch; ++j){
- copy_ongpu(input_size, l.delta_gpu + offset + j*l.outputs, 1, delta + j*input_size, 1);
+ axpy_ongpu(input_size, 1, l.delta_gpu + offset + j*l.outputs, 1, delta + j*input_size, 1);
}
offset += input_size;
}
--
Gitblit v1.10.0