mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Another CUDA performance improvements

Browse Source
This commit is contained in:
AlexeyAB
2019-01-18 16:29:54 +03:00
parent 5343aa4235
commit bf6b40f4e9
11 changed files with 94 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/network.c
View File

@ -856,6 +856,10 @@ void free_network(network net)
if (gpu_index >= 0) cuda_free(net.workspace);
else free(net.workspace);
if (net.input_state_gpu) cuda_free(net.input_state_gpu);
if (net.input_pinned_cpu) { // CPU
if (net.input_pinned_cpu_flag) cudaFreeHost(net.input_pinned_cpu);
else free(net.input_pinned_cpu);
}
if (*net.input_gpu) cuda_free(*net.input_gpu);
if (*net.truth_gpu) cuda_free(*net.truth_gpu);
if (net.input_gpu) free(net.input_gpu);