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

Fixed Python detection

Browse Source
This commit is contained in:
AlexeyAB
2018-04-17 14:21:23 +03:00
parent 140333977c
commit 6e3afec85a
4 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/cuda.h
View File

@ -31,11 +31,11 @@ extern int gpu_index;
#ifdef CUDNN
#include "cudnn.h"
#endif
#endif // CUDNN
#ifdef __cplusplus
extern "C" {
#endif
#endif // __cplusplus
void check_error(cudaError_t status);
cublasHandle_t blas_handle();
float *cuda_make_array(float *x, size_t n);
@ -51,12 +51,14 @@ extern "C" {
cudaStream_t get_cuda_stream();
#ifdef __cplusplus
}
#endif
#endif // __cplusplus
#ifdef CUDNN
cudnnHandle_t cudnn_handle();
enum {cudnn_fastest, cudnn_smallest};
#endif
#endif
#endif
#else // GPU
YOLODLL_API void cuda_set_device(int n);
#endif // GPU
#endif // CUDA_H