diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/multinomial_impl.cu b/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/multinomial_impl.cu index c1cbefe31a83386805477544d428a847486a3887..2877413963e14e6c718b5ac3895a132b6566e312 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/multinomial_impl.cu +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/multinomial_impl.cu @@ -14,6 +14,7 @@ * limitations under the License. */ +#include #include "multinomial_impl.cuh" template @@ -107,7 +108,8 @@ void Multinomial(int seed, T *input, int num_sample, curandState *globalState, i if (seed != 0) { RNG_seed = seed; } else { - RNG_seed = time(NULL); + std::random_device rd; + RNG_seed = static_cast(rd()); } int count = distributions * num_sample; MultinomialKernel<<>>(RNG_seed, input, num_sample, globalState,