m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modify dump task proto

Browse Source
pull/903/head
zhou_chao1993 4 years ago
parent 341eb10886
commit 6ba36558c4
6 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File

1
ge/executor/proto/dump_task.proto
Unescape View File

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

1
ge/graph/load/new_model_manager/data_dumper.cc
Unescape View File

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

4
ge/graph/load/new_model_manager/davinci_model.cc
Unescape View File

@ -545,7 +545,7 @@ Status DavinciModel::SetTSDevice() {
Status DavinciModel::OpDebugRegister() { Status DavinciModel::OpDebugRegister() {
bool is_op_debug = false; bool is_op_debug = false;
(void)ge::AttrUtils::GetBool(ge_model_, ATTR_OP_DEBUG_FLAG, is_op_debug); (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) { if (is_op_debug) {
debug_reg_mutex_.lock(); debug_reg_mutex_.lock();
rtError_t rt_ret = rtMalloc(&op_debug_addr_, kOpDebugMemorySize, RT_MEMORY_DDR); 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(); version_ = ge_model_->GetVersion();
name_ = ge_model_->GetName(); name_ = ge_model_->GetName();
(void)ge::AttrUtils::GetBool(ge_model_, ATTR_NAME_SWITCH_FOR_L1_FUSION, is_l1_fusion_enable_); (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(); CheckHasHcomOp();
vector<int64_t> huge_stream_list; vector<int64_t> huge_stream_list;

1
ge/proto/dump_task.proto
Unescape View File

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

2
metadef

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

2
parser

@ -1 +1 @@
Subproject commit 8ba13b54160440e13e7c10d5d1f04c98d91e27c7 Subproject commit 4e72aae41e78af1a19cd965da4a45cbd988b9a75
Write Preview
Loading…
Cancel
Save