p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!47625 fix renorm gpu kernel bug 

Merge pull request !47625 from liubuyu/bug_fix
This commit is contained in:
i-robot 2023-01-16 09:41:12 +00:00 committed by Gitee
parent f8f8d33564 6a62035ca9
commit f5c5189f35
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/plugin/device/gpu/kernel/math/renorm_gpu_kernel.cc
View File

@ -106,6 +106,10 @@ void RenormGpuKernelMod::InitParams() {
if (dim_ < 0) {
dim_ += shape_size;
}
axis_size_ = 1;
inner_size_ = 1;
stride_size_ = 1;
total_size_ = 1;
for (size_t i = 0; i < shape_size; ++i) {
if (SizeToLong(i) == dim_) {
axis_size_ *= input_shape_[i];