Browse Source

!1469 modify dynamic shape dump in c77

From: @zhou_chao1993
Reviewed-by: @ji_chen,@xchu42
Signed-off-by: @lbisdaddy
tags/v1.2.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
089b82e9bd
1 changed files with 7 additions and 5 deletions
  1. +7
    -5
      ge/hybrid/executor/worker/execution_engine.cc

+ 7
- 5
ge/hybrid/executor/worker/execution_engine.cc View File

@@ -233,11 +233,13 @@ Status NodeDoneCallback::DumpDynamicNode() {
uintptr_t output_addr = reinterpret_cast<uintptr_t>(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<void *>(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<void *>(varible_loop_per_iter->GetData());
}

void *loop_cond = nullptr;
TensorValue *varible_loop_cond = context_->GetVariable(NODE_NAME_FLOWCTRL_LOOP_COND);


Loading…
Cancel
Save