Browse Source

modify dump task proto

tags/v1.2.0
zhou_chao1993 3 years ago
parent
commit
6ba36558c4
6 changed files with 7 additions and 4 deletions
  1. +1
    -0
      ge/executor/proto/dump_task.proto
  2. +1
    -0
      ge/graph/load/new_model_manager/data_dumper.cc
  3. +2
    -2
      ge/graph/load/new_model_manager/davinci_model.cc
  4. +1
    -0
      ge/proto/dump_task.proto
  5. +1
    -1
      metadef
  6. +1
    -1
      parser

+ 1
- 0
ge/executor/proto/dump_task.proto View File

@@ -109,4 +109,5 @@ message DumpData{
repeated OpOutput output = 3;
repeated OpInput input = 4;
repeated OpBuffer buffer = 5;
string op_name = 6;
}

+ 1
- 0
ge/graph/load/new_model_manager/data_dumper.cc View File

@@ -905,6 +905,7 @@ Status DataDumper::DumpExceptionInfo(const std::vector<rtExceptionInfo> exceptio
toolkit::dumpdata::DumpData dump_data;
dump_data.set_version("2.0");
dump_data.set_dump_time(GetNowTime());
dump_data.set_op_name(op_desc_info.op_name);
for (size_t i = 0; i < op_desc_info.input_format.size(); ++i) {
toolkit::dumpdata::OpInput input;
input.set_data_type(toolkit::dumpdata::OutputDataType(GetIrDataType(op_desc_info.input_data_type[i])));


+ 2
- 2
ge/graph/load/new_model_manager/davinci_model.cc View File

@@ -545,7 +545,7 @@ Status DavinciModel::SetTSDevice() {
Status DavinciModel::OpDebugRegister() {
bool is_op_debug = false;
(void)ge::AttrUtils::GetBool(ge_model_, ATTR_OP_DEBUG_FLAG, is_op_debug);
GELOGD("The value of op_debug in ge_model_ is %d.", is_op_debug);
GELOGD("The value of op debug in ge_model is %d.", is_op_debug);
if (is_op_debug) {
debug_reg_mutex_.lock();
rtError_t rt_ret = rtMalloc(&op_debug_addr_, kOpDebugMemorySize, RT_MEMORY_DDR);
@@ -638,7 +638,7 @@ Status DavinciModel::Init(void *dev_ptr, size_t mem_size, void *weight_ptr, size
version_ = ge_model_->GetVersion();
name_ = ge_model_->GetName();
(void)ge::AttrUtils::GetBool(ge_model_, ATTR_NAME_SWITCH_FOR_L1_FUSION, is_l1_fusion_enable_);
GELOGD("The value of ge.l1Fusion in ge_model_ is %d.", is_l1_fusion_enable_);
GELOGD("The value of ge.l1Fusion in ge_model is %d.", is_l1_fusion_enable_);
CheckHasHcomOp();

vector<int64_t> huge_stream_list;


+ 1
- 0
ge/proto/dump_task.proto View File

@@ -109,4 +109,5 @@ message DumpData{
repeated OpOutput output = 3;
repeated OpInput input = 4;
repeated OpBuffer buffer = 5;
string op_name = 6;
}

+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 4d69fdedebfad0ecfaacc7955cb87a547d5fbf14
Subproject commit dc6cceb67bc82b567bcbd6f415776644253e1467

+ 1
- 1
parser

@@ -1 +1 @@
Subproject commit 8ba13b54160440e13e7c10d5d1f04c98d91e27c7
Subproject commit 4e72aae41e78af1a19cd965da4a45cbd988b9a75

Loading…
Cancel
Save