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;