Browse Source

modify deleter

pull/1471/head
zhou_chao1993 4 years ago
parent
commit
b16d58e465
3 changed files with 16 additions and 19 deletions
  1. +8
    -8
      ge/graph/manager/graph_manager.cc
  2. +7
    -10
      ge/graph/manager/host_mem_manager.cc
  3. +1
    -1
      metadef

+ 8
- 8
ge/graph/manager/graph_manager.cc View File

@@ -222,7 +222,7 @@ Status GraphManager::UnloadModel(GeRootModelPtr ge_root_model, uint32_t graph_id
Status ret = SUCCESS;
for (size_t i = 0; i < ge_root_model->GetAllModelId().size(); ++i) {
uint32_t model_id = ge_root_model->GetAllModelId()[i];
GELOGI("Unload model %u.", model_id);
GELOGI("Unload model %u", model_id);
ret = GraphLoader::UnloadModel(model_id);
if (ret != SUCCESS) {
GELOGW("[GraphManager] unload model failed, modelId=%u, graphId=%u.", model_id, graph_id);
@@ -324,7 +324,7 @@ Status GraphManager::InitDynamicParams(ComputeGraphPtr &compute_graph) {
return FAILED;
}
if ((op_desc->GetType() == DATA) || (op_type == kGetNextName)) {
GELOGI("Need to process multi batch for compute graph. op_type:%s.", op_desc->GetType().c_str());
GELOGI("Need to process multi batch for compute graph, op_type:%s.", op_desc->GetType().c_str());
GetLocalOmgContext().need_multi_batch = true;
break;
}
@@ -349,7 +349,7 @@ Status GraphManager::InitDynamicParams(ComputeGraphPtr &compute_graph) {
void GraphManager::SetAddGraphCondition(GraphId graph_id, uint32_t cond) {
std::lock_guard<std::mutex> lock(add_graph_cond_mutex_);
graph_id_to_add_graph_cond_[graph_id] = cond;
GELOGD("Graph [id:%u] has been added.", graph_id);
GELOGD("Graph [id:%u] has been added", graph_id);
}

uint32_t GraphManager::GetAddGraphCondition(GraphId graph_id) {
@@ -358,7 +358,7 @@ uint32_t GraphManager::GetAddGraphCondition(GraphId graph_id) {
if (it != graph_id_to_add_graph_cond_.end()) {
return it->second;
} else {
GELOGD("Graph [id:%u] has not been added.", graph_id);
GELOGD("Graph [id:%u] has not been added", graph_id);
return kNotAdded;
}
}
@@ -368,9 +368,9 @@ void GraphManager::RemoveAddGraphCondition(GraphId graph_id) {
auto it = graph_id_to_add_graph_cond_.find(graph_id);
if (it != graph_id_to_add_graph_cond_.end()) {
graph_id_to_add_graph_cond_.erase(it);
GELOGD("Successfully removed add_graph_cond of graph [id:%u].", graph_id);
GELOGD("Successfully remove add_graph_cond of graph [id:%u].", graph_id);
} else {
GELOGD("Graph [id:%u] has not been added. no need to remove.", graph_id);
GELOGD("Graph [id:%u] has not been added, no need to be removed.", graph_id);
}
}

@@ -390,7 +390,7 @@ Status GraphManager::CheckRepeatAdd(uint32_t graph_id, bool &is_added) {
GraphNodePtr graph_node;
Status ret = GetGraphNode(graph_id, graph_node);
if (ret != SUCCESS) {
GELOGE(ret, "[AddGraph] GetGraphNode failed, graph_id = %u.", graph_id);
GELOGE(ret, "[AddGraph] GetGraphNode failed, graph_id:%u.", graph_id);
return ret;
}
is_added = true;
@@ -418,7 +418,7 @@ Status GraphManager::NotifyWaittingGraph(uint32_t graph_id) {
GELOGE(INTERNAL_ERROR, "Get graph [id:%u] count failed, graph might have not been added.", graph_id);
return INTERNAL_ERROR;
}
GELOGD("Add graph finished, graph_id:%u", graph_id);
GELOGD("Add graph finished, graph_id is :%u", graph_id);
if (count > 1) {
GELOGD("Finish addgraph, graph_id:%u, graph_count:%u, start to notify.", graph_id, count);
add_graph_cv_.notify_all();


+ 7
- 10
ge/graph/manager/host_mem_manager.cc View File

@@ -45,12 +45,11 @@ Status SharedMemAllocator::Allocate(SharedMemInfo &mem_info) {
return GE_GRAPH_MEMORY_ALLOC_FAILED;
}
mem_info.fd = output_para.fd;
mem_info.host_aligned_ptr = AlignedPtr::BuildFromAllocFunc([&output_para](std::unique_ptr<uint8_t[], deleter> &ptr) {
ptr.reset(reinterpret_cast<uint8_t *>(output_para.ptr));
},
[](uint8_t *ptr) {
ptr = nullptr;
});
mem_info.host_aligned_ptr = AlignedPtr::BuildFromAllocFunc(
[&output_para](std::unique_ptr<uint8_t[], AlignedPtr::Deleter> &ptr) {
ptr.reset(reinterpret_cast<uint8_t *>(output_para.ptr));
},
[](uint8_t *ptr) { ptr = nullptr; });
mem_info.device_address = reinterpret_cast<uint8_t *>(output_para.devPtr);
return SUCCESS;
}
@@ -99,8 +98,7 @@ Status HostMemManager::MallocSharedMemory(SharedMemInfo &mem_info) {
std::lock_guard<std::recursive_mutex> lock(mutex_);
auto iter = var_memory_base_map_.find(mem_info.op_name);
if (iter != var_memory_base_map_.end()) {
REPORT_INNER_ERROR("E19999", "MemInfo.op_name:%s can't find in var_memory_base_map_",
mem_info.op_name.c_str());
REPORT_INNER_ERROR("E19999", "MemInfo.op_name:%s can't find in var_memory_base_map_", mem_info.op_name.c_str());
GELOGE(FAILED, "Host shared memory for op %s has been malloced", mem_info.op_name.c_str());
return FAILED;
}
@@ -114,8 +112,7 @@ Status HostMemManager::MallocSharedMemory(SharedMemInfo &mem_info) {
Status HostMemManager::QueryVarMemInfo(const string &op_name, uint64_t &base_addr, uint64_t &data_size) {
std::lock_guard<std::recursive_mutex> lock(mutex_);
if (var_memory_base_map_.find(op_name) == var_memory_base_map_.end()) {
REPORT_INNER_ERROR("E19999", "MemInfo.op_name:%s can't find in var_memory_base_map_",
op_name.c_str());
REPORT_INNER_ERROR("E19999", "MemInfo.op_name:%s can't find in var_memory_base_map_", op_name.c_str());
GELOGE(INTERNAL_ERROR, "Find host base base_addr failed,node name:%s!", op_name.c_str());
return INTERNAL_ERROR;
}


+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 22ab76ecd8461f679606374be4b3b6b6f7cad321
Subproject commit 1c41e02f73b6e8f95369e052ee4de285145fb34f

Loading…
Cancel
Save