From e205c1e7aeb47e3dffd35d1b5ce7841d24b9aff4 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 04 Jan 2018 23:23:10 +0000 Subject: [PATCH] Tracking fixed --- src/cuda.h | 5 +++++ 1 files changed, 5 insertions(+), 0 deletions(-) diff --git a/src/cuda.h b/src/cuda.h index 29b1eef..31f9092 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 @@ -26,6 +30,7 @@ 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