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

Added CUDA-streams to Darknet-Yolo forward inference

Browse Source
This commit is contained in:
AlexeyAB
2017-12-31 20:10:32 +03:00
parent 83d9b0b6d8
commit 9d23aad869
10 changed files with 66 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/cuda.h
View File

@ -30,6 +30,7 @@ void cuda_free(float *x_gpu);
void cuda_random(float *x_gpu, size_t n);
float cuda_compare(float *x_gpu, float *x, size_t n, char *s);
dim3 cuda_gridsize(size_t n);
cudaStream_t get_cuda_stream();
#ifdef CUDNN
cudnnHandle_t cudnn_handle();