From 8672fe4afb975409df3a20193e08161761c3af06 Mon Sep 17 00:00:00 2001
From: hanhuifeng2020 <hanhuifeng1@huawei.com>
Date: Thu, 15 Sep 2022 16:14:03 +0800
Subject: [PATCH] fix some bugs of ynamicBroadcastGradientArgsCpuKernelMod code

---
 .../cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.cc  | 3 +--
 .../cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.h   | 8 +-------
 2 files changed, 2 insertions(+), 9 deletions(-)

diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.cc
index 3b8603a7e3e..9ce89b8eccb 100644
--- a/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.cc
+++ b/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.cc
@@ -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);
 }
diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.h b/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.h
index 4d5f01e8333..b923ef9ecd7 100644
--- a/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.h
+++ b/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_broadcast_grad_args_cpu_kernel.h
@@ -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_)};