From bb1db5e868ead75ec7c96949c19d504c0f500c1a Mon Sep 17 00:00:00 2001 From: zhaozhixuan Date: Fri, 26 Mar 2021 14:21:16 +0800 Subject: [PATCH] Fix bug of memcpy input 0. --- .../executor/hybrid_model_async_executor.cc | 25 ++++++++++--------- 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/ge/hybrid/executor/hybrid_model_async_executor.cc b/ge/hybrid/executor/hybrid_model_async_executor.cc index 454bb942..ead5f8d7 100644 --- a/ge/hybrid/executor/hybrid_model_async_executor.cc +++ b/ge/hybrid/executor/hybrid_model_async_executor.cc @@ -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;