From 729ce43e6ec45cfdb58e06e227428a0f81c5de0f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 10 Jun 2016 00:20:31 +0000
Subject: [PATCH] stuff
---
src/network.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/network.c b/src/network.c
index b617f7e..2960d67 100644
--- a/src/network.c
+++ b/src/network.c
@@ -434,7 +434,7 @@
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);
+ net->workspace = calloc(1, workspace_size);
#endif
//fprintf(stderr, " Done!\n");
return 0;
--
Gitblit v1.10.0