From 75a25c5e60784ac4a87950e2b1c2b506f4d7b17b Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 29 Jun 2017 17:52:42 +0000 Subject: [PATCH] For the console-app that uses DLL changed: MT-Debug-DLL (/MDd) to MT-DLL (/MD) --- src/cuda.h | 25 +++++++++++++++++++------ 1 files changed, 19 insertions(+), 6 deletions(-) diff --git a/src/cuda.h b/src/cuda.h index 9d949b0..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 @@ -11,16 +15,25 @@ #include "curand.h" #include "cublas_v2.h" +#ifdef CUDNN +#include "cudnn.h" +#endif + void check_error(cudaError_t status); cublasHandle_t blas_handle(); -float *cuda_make_array(float *x, int n); -int *cuda_make_int_array(int n); -void cuda_push_array(float *x_gpu, float *x, int n); -void cuda_pull_array(float *x_gpu, float *x, int n); +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); +void cuda_set_device(int n); void cuda_free(float *x_gpu); -void cuda_random(float *x_gpu, int n); -float cuda_compare(float *x_gpu, float *x, int n, char *s); +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); +#ifdef CUDNN +cudnnHandle_t cudnn_handle(); +#endif + #endif #endif -- Gitblit v1.10.0