From c021841a52a7b71224b755bfbf779b05012b59d5 Mon Sep 17 00:00:00 2001
From: Tino Hager <tino.hager@nager.at>
Date: Tue, 10 Jul 2018 19:46:14 +0000
Subject: [PATCH] Get gpu count and name
---
src/network.c | 9 +++++++--
1 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/src/network.c b/src/network.c
index b32dfc0..d135a29 100644
--- a/src/network.c
+++ b/src/network.c
@@ -443,8 +443,8 @@
}
#ifdef GPU
if(gpu_index >= 0){
- printf(" try to allocate workspace = %zu * sizeof(float), ", (workspace_size - 1) / sizeof(float) + 1);
- net->workspace = cuda_make_array(0, (workspace_size-1)/sizeof(float)+1);
+ printf(" try to allocate workspace = %zu * sizeof(float), ", workspace_size / sizeof(float) + 1);
+ net->workspace = cuda_make_array(0, workspace_size/sizeof(float) + 1);
printf(" CUDA allocate done! \n");
}else {
free(net->workspace);
@@ -493,6 +493,11 @@
return def;
}
+layer* get_network_layer(network* net, int i)
+{
+ return net->layers + i;
+}
+
image get_network_image(network net)
{
int i;
--
Gitblit v1.10.0