From d7d7da2653ff4f79a275529b0ac3fec438880083 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 27 Mar 2015 02:13:59 +0000
Subject: [PATCH] Fixed im2col mistake >< face#palm
---
src/cuda.c | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/src/cuda.c b/src/cuda.c
index 7982953..fb7485e 100644
--- a/src/cuda.c
+++ b/src/cuda.c
@@ -66,6 +66,7 @@
if(!init){
curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_DEFAULT);
curandSetPseudoRandomGeneratorSeed(gen, 0ULL);
+ init = 1;
}
curandGenerateUniform(gen, x_gpu, n);
check_error(cudaPeekAtLastError());
--
Gitblit v1.10.0