From cd8a3dcb4ca42f22ad8f46a95e00977c92be6bbd Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 08 Feb 2018 23:22:42 +0000 Subject: [PATCH] Compile fixes --- src/cuda.h | 6 ++++++ 1 files changed, 6 insertions(+), 0 deletions(-) diff --git a/src/cuda.h b/src/cuda.h index 29b1eef..31577ff 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 @@ -22,10 +26,12 @@ 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); +int cuda_get_device(); 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(); -- Gitblit v1.10.0