yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Changed CURAND generator to work on a wider set of platforms.

Browse Source
This commit is contained in:
Sylvain Jeaugey 2016-06-06 14:34:03 -07:00
parent 9d9d8cd59f
commit bd3cf73e6e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/include/test_utilities.h
View File

@ -109,7 +109,7 @@ void GenerateRandom<unsigned long long>(curandGenerator_t generator, unsigned lo
template<typename T> template<typename T>
void Randomize(T* const dest, const int N, const int randomSeed) { void Randomize(T* const dest, const int N, const int randomSeed) {
curandGenerator_t gen; curandGenerator_t gen;
CURAND_CHK(curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_MT19937)); CURAND_CHK(curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_MTGP32));
CURAND_CHK(curandSetPseudoRandomGeneratorSeed(gen, randomSeed)); CURAND_CHK(curandSetPseudoRandomGeneratorSeed(gen, randomSeed));
GenerateRandom<T>(gen, dest, N); GenerateRandom<T>(gen, dest, N);
CURAND_CHK(curandDestroyGenerator(gen)); CURAND_CHK(curandDestroyGenerator(gen));
@ -144,7 +144,7 @@ __global__ void halve(const float * src, half* dest, int N) {
template<> template<>
void Randomize<half>(half* const dest, const int N, const int randomSeed) { void Randomize<half>(half* const dest, const int N, const int randomSeed) {
curandGenerator_t gen; curandGenerator_t gen;
CURAND_CHK(curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_MT19937)); CURAND_CHK(curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_MTGP32));
CURAND_CHK(curandSetPseudoRandomGeneratorSeed(gen, randomSeed)); CURAND_CHK(curandSetPseudoRandomGeneratorSeed(gen, randomSeed));
float* temp; float* temp;