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

5
include/darknet.h
View File

@ -316,6 +316,8 @@ struct layer {
float *col_image;
float * delta;
float * output;
int delta_pinned;
int output_pinned;
float * loss;
float * squared;
float * norms;
@ -582,6 +584,8 @@ typedef struct network {
float *output_gpu;
float *input_state_gpu;
float *input_pinned_cpu;
int input_pinned_cpu_flag;
float **input_gpu;
float **truth_gpu;
@ -777,6 +781,7 @@ LIB_API pthread_t load_data_in_thread(load_args args);
// cuda.h
LIB_API void cuda_pull_array(float *x_gpu, float *x, size_t n);
LIB_API void cuda_pull_array_async(float *x_gpu, float *x, size_t n);
LIB_API void cuda_set_device(int n);
// utils.h