Browse Source

reformat

tags/v1.1.0
chuxing 3 years ago
parent
commit
f9793741ff
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      ge/hybrid/executor/worker/execution_engine.cc

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

@@ -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());


Loading…
Cancel
Save