From 07b39c5f57665cbc4e806c6dd4cac552e1179f35 Mon Sep 17 00:00:00 2001 From: PTS93 <timon@skerutsch.de> Date: Sun, 28 Jan 2018 13:28:30 +0000 Subject: [PATCH] Fix autosave interval --- src/cuda.h | 7 +++++++ 1 files changed, 7 insertions(+), 0 deletions(-) diff --git a/src/cuda.h b/src/cuda.h index cdd6db9..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 @@ -21,10 +25,13 @@ 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); +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