From 32d2c969973aa98635123743f321859192ff581d Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 09 Jun 2016 22:09:30 +0000
Subject: [PATCH] hiding secret broken things
---
src/network.c | 11 +++++++++++
1 files changed, 11 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index 8f39f7b..b617f7e 100644
--- a/src/network.c
+++ b/src/network.c
@@ -137,6 +137,7 @@
void forward_network(network net, network_state state)
{
+ state.workspace = net.workspace;
int i;
for(i = 0; i < net.n; ++i){
state.index = i;
@@ -252,6 +253,7 @@
int i;
float *original_input = state.input;
float *original_delta = state.delta;
+ state.workspace = net.workspace;
for(i = net.n-1; i >= 0; --i){
state.index = i;
if(i == 0){
@@ -400,6 +402,7 @@
net->w = w;
net->h = h;
int inputs = 0;
+ size_t workspace_size = 0;
//fprintf(stderr, "Resizing to %d x %d...", w, h);
//fflush(stderr);
for (i = 0; i < net->n; ++i){
@@ -419,12 +422,20 @@
}else{
error("Cannot resize this type of layer");
}
+ if(l.workspace_size > workspace_size) workspace_size = l.workspace_size;
inputs = l.outputs;
net->layers[i] = l;
w = l.out_w;
h = l.out_h;
if(l.type == AVGPOOL) break;
}
+#ifdef GPU
+ cuda_free(net->workspace);
+ net->workspace = cuda_make_array(0, (workspace_size-1)/sizeof(float)+1);
+#else
+ free(net->workspace);
+ net->workspace = calloc(1, (workspace_size-1)/sizeof(float)+1);
+#endif
//fprintf(stderr, " Done!\n");
return 0;
}
--
Gitblit v1.10.0