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

fix some bugs of ynamicBroadcastGradientArgsCpuKernelMod code

This commit is contained in:
hanhuifeng2020 2022-09-15 16:14:03 +08:00
parent b1a32c6c89
commit 8672fe4afb
2 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.cc
View File

@ -163,7 +163,7 @@ bool DynamicBroadcastGradientArgsCpuKernelMod::Init(const BaseOperatorPtr &base_
if (!MatchKernelFunc(base_operator, inputs, outputs)) {
return false;
}
is_need_retrieve_output_shape_ = true;
return true;
}
@ -177,7 +177,6 @@ int DynamicBroadcastGradientArgsCpuKernelMod::Resize(const BaseOperatorPtr &base
}
// get input_shape
outputs_ = outputs;
is_need_retrieve_output_shape_ = true;
return static_cast<int>(KRET_OK);
}

8
mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.h
View File

@ -31,7 +31,7 @@ namespace kernel {
class DynamicBroadcastGradientArgsCpuKernelMod : public NativeCpuKernelMod,
public MatchKernelHelper<DynamicBroadcastGradientArgsCpuKernelMod> {
public:
DynamicBroadcastGradientArgsCpuKernelMod() : r0_size_(0), r1_size_(0) { ResetResource(); }
DynamicBroadcastGradientArgsCpuKernelMod() : r0_size_(0), r1_size_(0) {}
~DynamicBroadcastGradientArgsCpuKernelMod() override = default;
bool Init(const BaseOperatorPtr &base_operator, const std::vector<KernelTensorPtr> &inputs,
@ -51,12 +51,6 @@ class DynamicBroadcastGradientArgsCpuKernelMod : public NativeCpuKernelMod,
std::vector<KernelAttr> GetOpSupport() override { return OpSupport(); }
void ResetResource() noexcept {
input_size_list_.clear();
output_size_list_.clear();
workspace_size_list_.clear();
}
std::vector<KernelTensorPtr> GetOutputs() override {
ShapeVector r0_shape{SizeToLong(r0_size_)};
ShapeVector r1_shape{SizeToLong(r1_size_)};