From a4485b8a6656c2a2fa0b78ca7c035523c8149b8c Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Wed, 10 Jun 2015 15:49:25 +0000 Subject: [PATCH] Merge branch 'master' of pjreddie.com:jnet --- src/cuda.h | 2 ++ 1 files changed, 2 insertions(+), 0 deletions(-) diff --git a/src/cuda.h b/src/cuda.h index cbe7975..ff0d5f3 100644 --- a/src/cuda.h +++ b/src/cuda.h @@ -8,6 +8,7 @@ #define BLOCK 256 #include "cuda_runtime.h" +#include "curand.h" #include "cublas_v2.h" void check_error(cudaError_t status); @@ -17,6 +18,7 @@ void cuda_push_array(float *x_gpu, float *x, int n); void cuda_pull_array(float *x_gpu, float *x, 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); dim3 cuda_gridsize(size_t n); -- Gitblit v1.10.0