From f6200930b982163fc85e220da379a37424accdf4 Mon Sep 17 00:00:00 2001
From: zhaozhixuan <zhaozhixuan2@hisilicon.com>
Date: Fri, 26 Mar 2021 15:10:37 +0800
Subject: [PATCH] Fix bug of rtMemcpy input 0.

---
 .../executor/hybrid_model_async_executor.cc   | 24 ++++++++++---------
 1 file changed, 13 insertions(+), 11 deletions(-)

diff --git a/ge/hybrid/executor/hybrid_model_async_executor.cc b/ge/hybrid/executor/hybrid_model_async_executor.cc
index 454bb942..082dcab6 100644
--- a/ge/hybrid/executor/hybrid_model_async_executor.cc
+++ b/ge/hybrid/executor/hybrid_model_async_executor.cc
@@ -297,17 +297,19 @@ Status HybridModelAsyncExecutor::PrepareInputs(const InputData &current_data, Hy
                            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 > 0) {
+      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;