Relase extinfo ptr

5 years ago · 44c65d3b62
parent 03405224ad
commit 44c65d3b62
1 changed files with 4 additions and 1 deletions
--- a/src/ge/ge_runtime/task/aicpu_task.cc
+++ b/src/ge/ge_runtime/task/aicpu_task.cc
@ -37,7 +37,10 @@ AicpuTask::AicpuTask(const ModelContext &model_context, const std::shared_ptr<Ai
  }
 }

-AicpuTask::~AicpuTask() { ReleaseRtMem(&args_); }
+AicpuTask::~AicpuTask() {
+  ReleaseRtMem(&args_);
+  ReleaseRtMem(&ext_info_);
+}

 bool AicpuTask::Distribute() {
  GELOGI("InitAicpuTask start.");