From 4b366754716fc4b5d0148de03ee8d9710df1bfdb Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Sun, 12 Jul 2015 05:53:45 +0000 Subject: [PATCH] Merge branch 'master' of github.com:pjreddie/darknet --- 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