From dc9a618aad9295b89d2c652168a6e53ebdb0c78a Mon Sep 17 00:00:00 2001 From: weiyang Date: Thu, 5 Nov 2020 09:52:08 +0800 Subject: [PATCH] fix nullptr check --- .../load/new_model_manager/task_info/memcpy_async_task_info.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ge/graph/load/new_model_manager/task_info/memcpy_async_task_info.cc b/ge/graph/load/new_model_manager/task_info/memcpy_async_task_info.cc index 6eb53c8a..aa37dd07 100755 --- a/ge/graph/load/new_model_manager/task_info/memcpy_async_task_info.cc +++ b/ge/graph/load/new_model_manager/task_info/memcpy_async_task_info.cc @@ -95,7 +95,8 @@ Status MemcpyAsyncTaskInfo::Distribute() { } 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 uint32_t args_size = sizeof(void *) * 2; args_offset_ = davinci_model->GetTotalArgsSize();