Fix bug of rtMemcpy input 0.

pull/1369/head
unknown 4 years ago
parent 0667f28f38
commit 030e4a765e

@ -297,19 +297,20 @@ Status HybridModelAsyncExecutor::PrepareInputs(const InputData &current_data, Hy
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]", if (data_buf.length > 0) {
model_->root_runtime_param_.graph_id, GELOGI("[IMAS]CopyPlainData memcpy graph_%u type[F] output[%zu] memaddr[%p] mem_size[%zu] datasize[%lu]",
input_index, model_->root_runtime_param_.graph_id,
args.inputs[input_index].GetData(), input_index,
mem_size, args.inputs[input_index].GetData(),
data_buf.length); mem_size,
GE_CHK_RT_RET(rtMemcpy(args.inputs[input_index].MutableData(), data_buf.length);
mem_size, GE_CHK_RT_RET(rtMemcpy(args.inputs[input_index].MutableData(),
data_buf.data, mem_size,
data_buf.length, data_buf.data,
RT_MEMCPY_HOST_TO_DEVICE)); data_buf.length,
RT_MEMCPY_HOST_TO_DEVICE));
}
} }
return SUCCESS; return SUCCESS;
} }

Loading…
Cancel
Save