From 196f4d93c97cf92914d3631d19c91c4e55c2d848 Mon Sep 17 00:00:00 2001
From: unknown <zhaozhixuan2@hisilicon.com>
Date: Mon, 29 Mar 2021 15:23:26 +0800
Subject: [PATCH] rtMemcpy no input size is 0.

---
 ge/graph/load/model_manager/davinci_model.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ge/graph/load/model_manager/davinci_model.cc b/ge/graph/load/model_manager/davinci_model.cc
index 52642086..8f72be91 100755
--- a/ge/graph/load/model_manager/davinci_model.cc
+++ b/ge/graph/load/model_manager/davinci_model.cc
@@ -3293,7 +3293,7 @@ Status DavinciModel::UpdateIoTaskArgs(const std::map<uint32_t, ZeroCopyOffset> &
     void *basic_addr = data.second.GetBasicAddr();
     uint64_t data_size = data.second.GetDataSize();
     if (copy_only_addrs_.count(basic_addr) > 0) {
-      if (is_input) {
+      if (is_input && buffer.length > 0) {
         GELOGI("[IMAS] Find addr %p need direct copy from user malloc input %p", basic_addr, buffer.data);
         rtError_t rt_ret = rtMemcpy(basic_addr, data_size, buffer.data, buffer.length, RT_MEMCPY_DEVICE_TO_DEVICE);
         if (rt_ret != RT_ERROR_NONE) {