Browse Source

modify single op dump bug

tags/v1.2.0
zhou_chao1993 4 years ago
parent
commit
f49599b6c5
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      ge/common/dump/dump_op.cc

+ 4
- 0
ge/common/dump/dump_op.cc View File

@@ -204,6 +204,10 @@ Status DumpOp::ExecutorDumpOp(aicpu::dump::OpMappingInfo &op_mapping_info) {
} }


Status DumpOp::SetDumpModelName(aicpu::dump::OpMappingInfo &op_mapping_info) { Status DumpOp::SetDumpModelName(aicpu::dump::OpMappingInfo &op_mapping_info) {
if (dynamic_model_name_.empty() && dynamic_om_name_.empty()) {
GELOGI("Single op dump, no need set model name");
return SUCCESS;
}
std::set<std::string> model_list = dump_properties_.GetAllDumpModel(); std::set<std::string> model_list = dump_properties_.GetAllDumpModel();
bool not_find_by_omname = model_list.find(dynamic_om_name_) == model_list.end(); bool not_find_by_omname = model_list.find(dynamic_om_name_) == model_list.end();
bool not_find_by_modelname = model_list.find(dynamic_model_name_) == model_list.end(); bool not_find_by_modelname = model_list.find(dynamic_model_name_) == model_list.end();


Loading…
Cancel
Save