|
|
|
@@ -296,18 +296,19 @@ Status HybridModelAsyncExecutor::PrepareInputs(const InputData ¤t_data, Hy |
|
|
|
"input data size(%lu) does not match model required size(%lu), ret failed.", |
|
|
|
data_buf.length, |
|
|
|
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; |
|
|
|
|