Browse Source

!1175 del thread of Shrink

From: @yangyongqiang5033
Reviewed-by: @tangqunzhang,@ji_chen,@xchu42
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
317500557c
2 changed files with 1 additions and 6 deletions
  1. +1
    -5
      ge/graph/load/model_manager/davinci_model.cc
  2. +0
    -1
      ge/graph/load/model_manager/davinci_model.h

+ 1
- 5
ge/graph/load/model_manager/davinci_model.cc View File

@@ -765,7 +765,7 @@ Status DavinciModel::Init(void *dev_ptr, size_t mem_size, void *weight_ptr, size
}
}

CREATE_STD_THREAD(shrink_id_, &DavinciModel::Shrink, this);
Shrink();
return SUCCESS;
}

@@ -2722,10 +2722,6 @@ Status DavinciModel::DestroyThread() {
thread_id_.join();
}

if (shrink_id_.joinable()) {
shrink_id_.join();
}

return SUCCESS;
}



+ 0
- 1
ge/graph/load/model_manager/davinci_model.h View File

@@ -906,7 +906,6 @@ class DavinciModel {
vector<int64_t> output_memory_size_list_;

thread thread_id_;
thread shrink_id_;

shared_ptr<ModelListener> listener_;



Loading…
Cancel
Save