Browse Source

Fix bug of memcpy input 0.

pull/1358/MERGE^2
zhaozhixuan 4 years ago
parent
commit
bb1db5e868
1 changed files with 13 additions and 12 deletions
  1. +13
    -12
      ge/hybrid/executor/hybrid_model_async_executor.cc

+ 13
- 12
ge/hybrid/executor/hybrid_model_async_executor.cc View File

@@ -296,18 +296,19 @@ Status HybridModelAsyncExecutor::PrepareInputs(const InputData &current_data, Hy
"input data size(%lu) does not match model required size(%lu), ret failed.", "input data size(%lu) does not match model required size(%lu), ret failed.",
data_buf.length, data_buf.length,
mem_size); mem_size);

GELOGI("[IMAS]CopyPlainData memcpy graph_%u type[F] output[%zu] memaddr[%p] mem_size[%zu] datasize[%lu]",
model_->root_runtime_param_.graph_id,
input_index,
args.inputs[input_index].GetData(),
mem_size,
data_buf.length);
GE_CHK_RT_RET(rtMemcpy(args.inputs[input_index].MutableData(),
mem_size,
data_buf.data,
data_buf.length,
RT_MEMCPY_HOST_TO_DEVICE));
if (data_buf.length) {
GELOGI("[IMAS]CopyPlainData memcpy graph_%u type[F] output[%zu] memaddr[%p] mem_size[%zu] datasize[%lu]",
model_->root_runtime_param_.graph_id,
input_index,
args.inputs[input_index].GetData(),
mem_size,
data_buf.length);
GE_CHK_RT_RET(rtMemcpy(args.inputs[input_index].MutableData(),
mem_size,
data_buf.data,
data_buf.length,
RT_MEMCPY_HOST_TO_DEVICE));
}
} }


return SUCCESS; return SUCCESS;


Loading…
Cancel
Save