y00500818
3 years ago
2 changed files with 1 additions and 6 deletions
Unified View
Diff Options
-
+1 -5ge/graph/load/model_manager/davinci_model.cc
-
+0 -1ge/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; | return SUCCESS; | ||||
| } | } | ||||
| @@ -2722,10 +2722,6 @@ Status DavinciModel::DestroyThread() { | |||||
| thread_id_.join(); | thread_id_.join(); | ||||
| } | } | ||||
| if (shrink_id_.joinable()) { | |||||
| shrink_id_.join(); | |||||
| } | |||||
| return SUCCESS; | 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_; | vector<int64_t> output_memory_size_list_; | ||||
| thread thread_id_; | thread thread_id_; | ||||
| thread shrink_id_; | |||||
| shared_ptr<ModelListener> listener_; | shared_ptr<ModelListener> listener_; | ||||