yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!21989 Bugfix for RandomReal cpu kernel. 

Merge pull request !21989 from chengang/random_real_bug_fix
This commit is contained in:
i-robot 2021-08-18 09:36:44 +00:00 committed by Gitee
parent dde05c36b8 9d70eefb0e
commit d08e4ec03b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/kernel_compiler/cpu/random_cpu_kernel.cc
View File

@ -99,7 +99,7 @@ void LaunchUniformReal(unsigned int seed, const std::vector<AddressPtr> &inputs,
MS_LOG(EXCEPTION) << "Expect output number 1, actual got output number " << outputs.size();
}
// Init output address.
auto output = reinterpret_cast<int *>(outputs[0]->addr);
auto output = reinterpret_cast<float *>(outputs[0]->addr);
MS_EXCEPTION_IF_NULL(output);
// Init sample number.