!22895 fix precision error of resizebilineargrad

Merge pull request !22895 from Simson/opinfer
2021-09-06 13:40:04 +00:00 · 2021-09-06 13:40:04 +00:00 · e177a4981a
parent 5ed105ec01 8a0087bceb
commit e177a4981a
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/resize_bilinear_grad_gpu_kernel.h
+++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/resize_bilinear_grad_gpu_kernel.h
@ -89,7 +89,7 @@ class ResizeBilinearGradGpuKernel : public GpuKernel {
    for (auto x : dx_shape) {
      dx_size_ *= x;
    }
-    workspace_size_ = (dx_size_ / sizeof(T)) * sizeof(T);
+    workspace_size_ = (dx_size_ / sizeof(T)) * sizeof(float);
    align_corners_ = GetAttr<bool>(kernel_node, "align_corners");
    InitSizeLists();
    return true;