From c90cae141085992e2b08b7cdac101c02a4088cfa Mon Sep 17 00:00:00 2001 From: zhou_chao1993 Date: Wed, 7 Apr 2021 17:16:24 +0800 Subject: [PATCH] modify dynamic shape dump --- ge/hybrid/executor/worker/execution_engine.cc | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/ge/hybrid/executor/worker/execution_engine.cc b/ge/hybrid/executor/worker/execution_engine.cc index bb06acd8..f28490c3 100755 --- a/ge/hybrid/executor/worker/execution_engine.cc +++ b/ge/hybrid/executor/worker/execution_engine.cc @@ -233,11 +233,13 @@ Status NodeDoneCallback::DumpDynamicNode() { uintptr_t output_addr = reinterpret_cast(tensor_value->GetData()); output_addrs.emplace_back(output_addr); } - void *loop_per_iter = nullptr; - TensorValue *varible_loop_per_iter = context_->GetVariable(NODE_NAME_FLOWCTRL_LOOP_PER_ITER); - if (varible_loop_per_iter != nullptr) { - loop_per_iter = const_cast(varible_loop_per_iter->GetData()); - } + dump_op_.SetDumpInfo(context_->GetDumpProperties(), op_desc, input_addrs, output_addrs, stream); + + void *loop_per_iter = nullptr; + TensorValue *varible_loop_per_iter = context_->GetVariable(NODE_NAME_FLOWCTRL_LOOP_PER_ITER); + if (varible_loop_per_iter != nullptr) { + loop_per_iter = const_cast(varible_loop_per_iter->GetData()); + } void *loop_cond = nullptr; TensorValue *varible_loop_cond = context_->GetVariable(NODE_NAME_FLOWCTRL_LOOP_COND);