mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!45756 Fix bug for grid sample GPU 

Merge pull request !45756 from zhangyongxian/dev_zhangyongxian_gridsample
This commit is contained in:
i-robot 2022-11-24 02:18:00 +00:00 committed by Gitee
parent 70e1f39e9c 62164fc1b7
commit 8ff9f4d023
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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/plugin/device/gpu/kernel/nn/grid_sampler_gpu_kernel.h
View File

@ -113,7 +113,7 @@ class GridSampler2DGpuKernelMod : public NativeGpuKernelMod {
stride_compute(input_stride_, input_shape_);
stride_compute(grid_stride_, grid_shape_);
stride_compute(output_stride_, output_shape_);
size_ = input_shape_[kIndex0] * grid_shape_[kIndex1] * grid_shape_[kIndex2] * grid_shape_[kIndex3];
size_ = input_shape_[kIndex0] * grid_shape_[kIndex1] * grid_shape_[kIndex2];
return KRET_OK;
}