From bd3cf73e6e7bda21ab90174e8f1d34a1367ea50d Mon Sep 17 00:00:00 2001 From: Sylvain Jeaugey Date: Mon, 6 Jun 2016 14:34:03 -0700 Subject: [PATCH] Changed CURAND generator to work on a wider set of platforms. --- test/include/test_utilities.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/include/test_utilities.h b/test/include/test_utilities.h index 4196801..84218e8 100644 --- a/test/include/test_utilities.h +++ b/test/include/test_utilities.h @@ -109,7 +109,7 @@ void GenerateRandom(curandGenerator_t generator, unsigned lo template void Randomize(T* const dest, const int N, const int randomSeed) { curandGenerator_t gen; - CURAND_CHK(curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_MT19937)); + CURAND_CHK(curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_MTGP32)); CURAND_CHK(curandSetPseudoRandomGeneratorSeed(gen, randomSeed)); GenerateRandom(gen, dest, N); CURAND_CHK(curandDestroyGenerator(gen)); @@ -144,7 +144,7 @@ __global__ void halve(const float * src, half* dest, int N) { template<> void Randomize(half* const dest, const int N, const int randomSeed) { curandGenerator_t gen; - CURAND_CHK(curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_MT19937)); + CURAND_CHK(curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_MTGP32)); CURAND_CHK(curandSetPseudoRandomGeneratorSeed(gen, randomSeed)); float* temp;