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

!45748 fix a bug of conv2d_grad_input_gpu_kernel 

Merge pull request !45748 from hanhuifeng/fix_conv_grad_input_bug
This commit is contained in:
i-robot 2022-11-21 01:48:38 +00:00 committed by Gitee
parent 830341bce3 53b3527b0a
commit 9a0a3bd4a9
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/plugin/device/gpu/kernel/nn/conv2d_grad_input_gpu_kernel.cc
View File

@ -407,7 +407,6 @@ void ConvGradInputBkwGpuKernelMod::ResetResource() noexcept {
padded_size_ = 0;
workspace_size_ = 0;
use_pad_ = false;
beta_ = 0;
input_size_list_.clear();
output_size_list_.clear();
workspace_size_list_.clear();

2
mindspore/ccsrc/plugin/device/gpu/kernel/nn/conv2d_grad_input_gpu_kernel.h
View File

@ -116,7 +116,7 @@ class ConvGradInputBkwGpuKernelMod : public NativeGpuKernelMod, public MatchKern
size_t padded_size_;
size_t workspace_size_;
bool use_pad_;
float beta_;
float beta_{0};
bool get_dynamic_attr_value_{false};
std::vector<int64_t> input_shape_;
static constexpr size_t kShapeIndex_{2};