multi_task for single_op.

4 years ago · ac62ccf90e
parent 84845edb61
commit ac62ccf90e
1 changed files with 2 additions and 2 deletions
--- a/ge/hybrid/node_executor/task_context.cc
+++ b/ge/hybrid/node_executor/task_context.cc
@ -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) {