diff --git a/ge/hybrid/executor/worker/execution_engine.cc b/ge/hybrid/executor/worker/execution_engine.cc index fad899c6..d230b949 100755 --- a/ge/hybrid/executor/worker/execution_engine.cc +++ b/ge/hybrid/executor/worker/execution_engine.cc @@ -122,10 +122,10 @@ Status NodeDoneCallback::PrepareConstInputs(const NodeItem &node_item) { output_tensor->GetSize()); if (tensor_size > 0) { GE_CHK_RT_RET(rtMemcpy(host_buffer.data(), - tensor_size, - output_tensor->GetData(), - tensor_size, - RT_MEMCPY_DEVICE_TO_HOST)); + tensor_size, + output_tensor->GetData(), + tensor_size, + RT_MEMCPY_DEVICE_TO_HOST)); } tensor.SetData(std::move(host_buffer)); string session_id = std::to_string(context_->GetSessionId());