From 0dff437a692a5f875dd0293c628ac9172f697c69 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Mon, 28 Mar 2016 02:10:10 +0000 Subject: [PATCH] fixed old --- src/cuda.h | 4 +++- 1 files changed, 3 insertions(+), 1 deletions(-) diff --git a/src/cuda.h b/src/cuda.h index cbe7975..9d949b0 100644 --- a/src/cuda.h +++ b/src/cuda.h @@ -5,9 +5,10 @@ #ifdef GPU -#define BLOCK 256 +#define BLOCK 512 #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