Expose YOLODLL_API void cuda_pull_array()

vinjn

2018-07-04 4b9120105199c1803b13fb659935ff78378448a8

 src/cuda.h

@@ -31,17 +31,17 @@

#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);
   int *cuda_make_int_array(size_t n);
   void cuda_push_array(float *x_gpu, float *x, size_t n);
   void cuda_pull_array(float *x_gpu, float *x, size_t n);
    YOLODLL_API void cuda_pull_array(float *x_gpu, float *x, size_t n);
   YOLODLL_API void cuda_set_device(int n);
   int cuda_get_device();
   void cuda_free(float *x_gpu);
@@ -51,12 +51,14 @@
   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