From 158bb1bee9951875dbe3474d84c6663431e18301 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 21 Oct 2014 21:49:18 +0000
Subject: [PATCH] softmax on gpu
---
src/connected_layer.c | 7 +++++++
1 files changed, 7 insertions(+), 0 deletions(-)
diff --git a/src/connected_layer.c b/src/connected_layer.c
index ba83dc3..b41ae91 100644
--- a/src/connected_layer.c
+++ b/src/connected_layer.c
@@ -108,6 +108,12 @@
#ifdef GPU
+void pull_connected_layer(connected_layer layer)
+{
+ cl_read_array(layer.weights_cl, layer.weights, layer.inputs*layer.outputs);
+ cl_read_array(layer.biases_cl, layer.biases, layer.outputs);
+}
+
void update_connected_layer_gpu(connected_layer layer)
{
axpy_ongpu(layer.outputs, layer.learning_rate, layer.bias_updates_cl, 1, layer.biases_cl, 1);
@@ -116,6 +122,7 @@
scal_ongpu(layer.inputs*layer.outputs, 1.-layer.learning_rate*layer.decay, layer.weights_cl, 1);
axpy_ongpu(layer.inputs*layer.outputs, layer.learning_rate, layer.weight_updates_cl, 1, layer.weights_cl, 1);
scal_ongpu(layer.inputs*layer.outputs, layer.momentum, layer.weight_updates_cl, 1);
+ pull_connected_layer(layer);
}
void forward_connected_layer_gpu(connected_layer layer, cl_mem input)
--
Gitblit v1.10.0