Browse Source

rtMemcpy no input size is 0.

tags/v1.3.0
unknown 4 years ago
parent
commit
196f4d93c9
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/load/model_manager/davinci_model.cc

+ 1
- 1
ge/graph/load/model_manager/davinci_model.cc View File

@@ -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) {


Loading…
Cancel
Save