Fix bug of memcpy input 0.

4 years ago · bb1db5e868
parent 96eaa5364d
commit bb1db5e868
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
@ -296,7 +296,7 @@ Status HybridModelAsyncExecutor::PrepareInputs(const InputData &current_data, Hy
                           "input data size(%lu) does not match model required size(%lu), ret failed.",
                           data_buf.length,
                           mem_size);
-
+    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,
@ -309,6 +309,7 @@ Status HybridModelAsyncExecutor::PrepareInputs(const InputData &current_data, Hy
                             data_buf.length,
                             RT_MEMCPY_HOST_TO_DEVICE));
    }
+  }

  return SUCCESS;
 }