From af4e4f92dc9e5da160eb6c6870a7b38b863f1c6c Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Tue, 28 Oct 2014 02:45:06 +0000 Subject: [PATCH] getting rid of sub_arrays, nvidia driver memory leak --- src/connected_layer.h | 1 + 1 files changed, 1 insertions(+), 0 deletions(-) diff --git a/src/connected_layer.h b/src/connected_layer.h index 19bcfa2..1e5b4a7 100644 --- a/src/connected_layer.h +++ b/src/connected_layer.h @@ -48,6 +48,7 @@ void forward_connected_layer_gpu(connected_layer layer, cl_mem input); void backward_connected_layer_gpu(connected_layer layer, cl_mem input, cl_mem delta); void update_connected_layer_gpu(connected_layer layer); +void push_connected_layer(connected_layer layer); #endif #endif -- Gitblit v1.10.0