diff options
| author | Sylvain Jeaugey <sjeaugey@nvidia.com> | 2016-06-07 00:34:03 +0300 |
|---|---|---|
| committer | Sylvain Jeaugey <sjeaugey@nvidia.com> | 2016-06-07 00:34:03 +0300 |
| commit | bd3cf73e6e7bda21ab90174e8f1d34a1367ea50d (patch) | |
| tree | 3653e18dd8fd32e51536403e22b511adca291911 | |
| parent | 9d9d8cd59fd4e9527ba4552aedf82285def7d3a4 (diff) | |
Changed CURAND generator to work on a wider set of platforms.
| -rw-r--r-- | test/include/test_utilities.h | 4 |
1 files 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<unsigned long long>(curandGenerator_t generator, unsigned lo template<typename T> 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<T>(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>(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; |