forked from mindspore-Ecosystem/mindspore
!27190 update random normal op seed logic
Merge pull request !27190 from zichun_ye/random_normal_seed
This commit is contained in:
commit
707e53910c
|
@ -81,23 +81,18 @@ class RandomOpGpuKernel : public GpuKernel {
|
||||||
switch (random_op_type_) {
|
switch (random_op_type_) {
|
||||||
case RANDOM_OP_NORMAL: {
|
case RANDOM_OP_NORMAL: {
|
||||||
float *mask_f = GetDeviceAddress<float>(outputs, 0);
|
float *mask_f = GetDeviceAddress<float>(outputs, 0);
|
||||||
if (!states_init_) {
|
std::random_device rd;
|
||||||
int RNG_seed = 0;
|
int RNG_seed = static_cast<int>(rd());
|
||||||
std::random_device rd;
|
if (seed2_ != 0) {
|
||||||
if (seed2_ != 0) {
|
RNG_seed = seed2_;
|
||||||
RNG_seed = seed2_;
|
} else if (seed_ != 0) {
|
||||||
} else if (seed_ != 0) {
|
RNG_seed = seed_;
|
||||||
RNG_seed = seed_;
|
|
||||||
} else {
|
|
||||||
RNG_seed = static_cast<int>(rd());
|
|
||||||
}
|
|
||||||
CHECK_CURAND_RET_WITH_EXCEPT(curandCreateGenerator(&mask_generator_, CURAND_RNG_PSEUDO_PHILOX4_32_10),
|
|
||||||
"Failed to create generator");
|
|
||||||
CHECK_CURAND_RET_WITH_EXCEPT(curandSetPseudoRandomGeneratorSeed(mask_generator_, RNG_seed),
|
|
||||||
"Failed to SetPseudoRandomGeneratorSeed");
|
|
||||||
MS_EXCEPTION_IF_NULL(mask_generator_);
|
|
||||||
states_init_ = true;
|
|
||||||
}
|
}
|
||||||
|
CHECK_CURAND_RET_WITH_EXCEPT(curandCreateGenerator(&mask_generator_, CURAND_RNG_PSEUDO_PHILOX4_32_10),
|
||||||
|
"Failed to create generator");
|
||||||
|
CHECK_CURAND_RET_WITH_EXCEPT(curandSetPseudoRandomGeneratorSeed(mask_generator_, RNG_seed),
|
||||||
|
"Failed to SetPseudoRandomGeneratorSeed");
|
||||||
|
MS_EXCEPTION_IF_NULL(mask_generator_);
|
||||||
CHECK_CURAND_RET_WITH_EXCEPT(curandSetStream(mask_generator_, reinterpret_cast<cudaStream_t>(stream_ptr)),
|
CHECK_CURAND_RET_WITH_EXCEPT(curandSetStream(mask_generator_, reinterpret_cast<cudaStream_t>(stream_ptr)),
|
||||||
"Failed to set stream for generator");
|
"Failed to set stream for generator");
|
||||||
// curandGen only support float or double for mask.
|
// curandGen only support float or double for mask.
|
||||||
|
|
Loading…
Reference in New Issue