!1175 del thread of Shrink

From: @yangyongqiang5033
Reviewed-by: @tangqunzhang,@ji_chen,@xchu42
Signed-off-by: @ji_chen
pull/1175/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 317500557c

@ -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;
}

@ -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