Expose YOLODLL_API void cuda_pull_array()

pull/1152/head
vinjn 7 years ago
parent 29407fa554
commit 4b91201051
  1. 2
      src/cuda.h

@ -41,7 +41,7 @@ extern "C" {
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);

Loading…
Cancel
Save