From 351996d9f0390ef44412fa678bc7a073a94e23e5 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 16 Mar 2017 18:49:36 +0000 Subject: [PATCH] Fixed memory leak in DLL, added load_image() & free_image(), added read_names_from_file() --- src/cuda.h | 5 +++++ 1 files changed, 5 insertions(+), 0 deletions(-) diff --git a/src/cuda.h b/src/cuda.h index cdd6db9..32aaabb 100644 --- a/src/cuda.h +++ b/src/cuda.h @@ -1,6 +1,10 @@ #ifndef CUDA_H #define CUDA_H +#if defined(_MSC_VER) && _MSC_VER < 1900 + #define inline __inline +#endif + extern int gpu_index; #ifdef GPU @@ -21,6 +25,7 @@ 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); +void cuda_set_device(int n); 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); -- Gitblit v1.10.0