Browse Source

multi_task for single_op.

tags/v1.2.0
unknown 4 years ago
parent
commit
ac62ccf90e
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/hybrid/node_executor/task_context.cc

+ 2
- 2
ge/hybrid/node_executor/task_context.cc View File

@@ -252,7 +252,7 @@ Status TaskContext::AllocateOutput(int index,
}

if (outputs_start_[index].GetSize() > 0) {
reMemset(outputs_start_[index].MutableData(), outputs_start_[index].GetSize(), 0, outputs_start_[index].GetSize());
rtMemset(outputs_start_[index].MutableData(), outputs_start_[index].GetSize(), 0, outputs_start_[index].GetSize());
}

if (execution_context_->trace_enabled) {
@@ -401,7 +401,7 @@ Status TaskContext::PropagateOutputs() {
subgraph_context_->all_inputs_[input_offset] = *tensor;
if (execution_context_->trace_enabled) {
subgraph_context_->all_inputs_[input_offset].SetName(
dst_node_item->NodeName() + "_in_" + std::to_string(dst_input_idx));
dst_node_item->NodeName() + "_in_" + std::to_string(dst_input_idx));
}
}
}


Loading…
Cancel
Save