Browse Source

!222 fix dynamic shape nullptr check

Merge pull request !222 from weiyang/development
tags/v1.1.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
1826d06f34
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      ge/graph/load/new_model_manager/task_info/memcpy_async_task_info.cc

+ 2
- 1
ge/graph/load/new_model_manager/task_info/memcpy_async_task_info.cc View File

@@ -95,7 +95,8 @@ Status MemcpyAsyncTaskInfo::Distribute() {
} }


Status MemcpyAsyncTaskInfo::CalculateArgs(const domi::TaskDef &task_def, DavinciModel *davinci_model) { Status MemcpyAsyncTaskInfo::CalculateArgs(const domi::TaskDef &task_def, DavinciModel *davinci_model) {
OpDescPtr op_desc = davinci_model_->GetOpByIndex(task_def.memcpy_async().op_index());
GE_CHECK_NOTNULL(davinci_model);
OpDescPtr op_desc = davinci_model->GetOpByIndex(task_def.memcpy_async().op_index());
// the num of src and dst size is 2 // the num of src and dst size is 2
uint32_t args_size = sizeof(void *) * 2; uint32_t args_size = sizeof(void *) * 2;
args_offset_ = davinci_model->GetTotalArgsSize(); args_offset_ = davinci_model->GetTotalArgsSize();


Loading…
Cancel
Save