From 38bd6ae6ba24fc8c14fd61d1238ae94a983434b3 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 20 Jul 2015 23:16:26 +0000
Subject: [PATCH] Better partial function
---
src/cuda.c | 4 +++-
1 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/src/cuda.c b/src/cuda.c
index 7982953..1b914a5 100644
--- a/src/cuda.c
+++ b/src/cuda.c
@@ -7,6 +7,7 @@
#include "blas.h"
#include "assert.h"
#include <stdlib.h>
+#include <time.h>
void check_error(cudaError_t status)
@@ -65,7 +66,8 @@
static int init = 0;
if(!init){
curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_DEFAULT);
- curandSetPseudoRandomGeneratorSeed(gen, 0ULL);
+ curandSetPseudoRandomGeneratorSeed(gen, time(0));
+ init = 1;
}
curandGenerateUniform(gen, x_gpu, n);
check_error(cudaPeekAtLastError());
--
Gitblit v1.10.0