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

fix gpu slicegrad

This commit is contained in:
fanjibin 2022-12-14 17:28:51 +08:00
parent bdeeb90233
commit c7014ad816
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/arrays/slice_grad_gpu_kernel.cc
View File

@ -136,6 +136,10 @@ int SliceGradGpuKernelMod::Resize(const BaseOperatorPtr &base_operator, const st
return ret;
}
dy_shape_.clear();
input_shape_.clear();
begin_.clear();
size_.clear();
if (TryGetIntValue(inputs, kBeginIndex_, kernel_name_, &begin_) &&
TryGetIntValue(inputs, kSizeIndex_, kernel_name_, &size_)) {
ProccessAttr(inputs);