From 11c72b1132feca7c1252ea01d02da4cb497e723f Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Thu, 11 Jun 2015 22:38:58 +0000 Subject: [PATCH] testing on one image --- 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