Fix bug of memcpy input 0.

pull/1358/MERGE^2
zhaozhixuan 4 years ago
parent 96eaa5364d
commit bb1db5e868

@ -296,7 +296,7 @@ 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);
if (data_buf.length) {
GELOGI("[IMAS]CopyPlainData memcpy graph_%u type[F] output[%zu] memaddr[%p] mem_size[%zu] datasize[%lu]", GELOGI("[IMAS]CopyPlainData memcpy graph_%u type[F] output[%zu] memaddr[%p] mem_size[%zu] datasize[%lu]",
model_->root_runtime_param_.graph_id, model_->root_runtime_param_.graph_id,
input_index, input_index,
@ -309,6 +309,7 @@ Status HybridModelAsyncExecutor::PrepareInputs(const InputData &current_data, Hy
data_buf.length, data_buf.length,
RT_MEMCPY_HOST_TO_DEVICE)); RT_MEMCPY_HOST_TO_DEVICE));
} }
}
return SUCCESS; return SUCCESS;
} }

Loading…
Cancel
Save