!1369 Fix bug of rtMemcpy input 0.

From: @zhao_zhixuan Reviewed-by: @xchu42,@ji_chen Signed-off-by: @ji_chen
4 years ago · b7dd89761a
parent 9635b0c9b9 030e4a765e
commit b7dd89761a
1 changed files with 13 additions and 12 deletions
--- a/ge/hybrid/executor/hybrid_model_async_executor.cc
+++ b/ge/hybrid/executor/hybrid_model_async_executor.cc
@ -297,6 +297,7 @@ Status HybridModelAsyncExecutor::PrepareInputs(const InputData &current_data, Hy
                           data_buf.length,
                           mem_size);

+    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,
@ -309,7 +310,7 @@ Status HybridModelAsyncExecutor::PrepareInputs(const InputData &current_data, Hy
                             data_buf.length,
                             RT_MEMCPY_HOST_TO_DEVICE));
    }
-
+  }
  return SUCCESS;
 }