xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix seed

This commit is contained in:
baihuawei 2020-09-09 15:00:26 +08:00
parent fc8bd0dd03
commit 8a101524da
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/multinomial_impl.cu
View File

@ -14,6 +14,7 @@
* limitations under the License.
*/
#include <random>
#include "multinomial_impl.cuh"
template <typename T>
@ -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<int>(rd());
}
int count = distributions * num_sample;
MultinomialKernel<<<GET_BLOCKS(count), GET_THREADS, 0, cuda_stream>>>(RNG_seed, input, num_sample, globalState,