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

multi_task for single_op.

Browse Source
pull/1044/head
unknown 4 years ago
parent 84845edb61
commit ac62ccf90e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
ge/hybrid/node_executor/task_context.cc
Unescape View File

@ -252,7 +252,7 @@ Status TaskContext::AllocateOutput(int index,
} }
if (outputs_start_[index].GetSize() > 0) { 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) { if (execution_context_->trace_enabled) {
@ -401,7 +401,7 @@ Status TaskContext::PropagateOutputs() {
subgraph_context_->all_inputs_[input_offset] = *tensor; subgraph_context_->all_inputs_[input_offset] = *tensor;
if (execution_context_->trace_enabled) { if (execution_context_->trace_enabled) {
subgraph_context_->all_inputs_[input_offset].SetName( 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));
} }
} }
} }

Write Preview
Loading…
Cancel
Save