diff --git a/ge/ge_runtime/task/aicpu_task.cc b/ge/ge_runtime/task/aicpu_task.cc index 5b3d8e82..cc07365d 100644 --- a/ge/ge_runtime/task/aicpu_task.cc +++ b/ge/ge_runtime/task/aicpu_task.cc @@ -30,6 +30,7 @@ AicpuTask::AicpuTask(const ModelContext &model_context, const std::shared_ptr= stream_list.size() || event_id >= event_list.size()) { GELOGW("stream_list size:%zu, stream_id:%u, event_list size:%zu, event_id:%u", stream_list.size(), stream_id, event_list.size(), event_id); + return; } stream_ = stream_list[stream_id]; event_ = event_list[event_id]; diff --git a/ge/ge_runtime/task/event_wait_task.cc b/ge/ge_runtime/task/event_wait_task.cc index 558c2a59..41383da0 100644 --- a/ge/ge_runtime/task/event_wait_task.cc +++ b/ge/ge_runtime/task/event_wait_task.cc @@ -26,6 +26,7 @@ EventWaitTask::EventWaitTask(const ModelContext &model_context, const std::share event_(nullptr) { if (task_info_ == nullptr) { GELOGW("task_info_ is null!"); + return; } auto stream_list = model_context.stream_list(); auto event_list = model_context.event_list(); @@ -34,6 +35,7 @@ EventWaitTask::EventWaitTask(const ModelContext &model_context, const std::share if (stream_id >= stream_list.size() || event_id >= event_list.size()) { GELOGW("stream_list size:%zu, stream_id:%u, event_list size:%zu, event_id:%u", stream_list.size(), stream_id, event_list.size(), event_id); + return; } stream_ = stream_list[stream_id]; event_ = event_list[event_id]; diff --git a/ge/ge_runtime/task/hccl_task.cc b/ge/ge_runtime/task/hccl_task.cc index 4d76e492..06165053 100644 --- a/ge/ge_runtime/task/hccl_task.cc +++ b/ge/ge_runtime/task/hccl_task.cc @@ -36,6 +36,7 @@ HcclTask::HcclTask(const ModelContext &model_context, const std::shared_ptr(model_context, task_info), task_info_(task_info), stream_(nullptr) { if (task_info_ == nullptr) { GELOGW("task_info_ is null!"); + return; } auto stream_list = model_context.stream_list(); uint32_t stream_id = task_info->stream_id(); @@ -31,6 +32,7 @@ MemcpyAsyncTask::MemcpyAsyncTask(const ModelContext &model_context, GELOGI("Stream list size:%zu, stream id:%u.", stream_list.size(), stream_id); if (stream_id >= stream_list.size()) { GELOGW("Stream id invalid"); + return; } stream_ = stream_list[stream_id]; } diff --git a/ge/ge_runtime/task/profiler_task.cc b/ge/ge_runtime/task/profiler_task.cc index 535c4f36..dbbc8b8b 100644 --- a/ge/ge_runtime/task/profiler_task.cc +++ b/ge/ge_runtime/task/profiler_task.cc @@ -23,12 +23,14 @@ ProfilerTask::ProfilerTask(const ModelContext &model_context, const std::shared_ : TaskRepeater(model_context, task_info), task_info_(task_info), stream_(nullptr) { if (task_info_ == nullptr) { GELOGW("task_info_ is null!"); + return; } auto stream_list = model_context.stream_list(); uint32_t stream_id = task_info->stream_id(); GELOGI("Stream list size:%zu, stream id:%u.", stream_list.size(), stream_id); if (stream_id >= stream_list.size()) { GELOGW("Stream id invalid"); + return; } stream_ = stream_list[stream_id]; } diff --git a/ge/ge_runtime/task/stream_active_task.cc b/ge/ge_runtime/task/stream_active_task.cc index d58b8942..ff57e377 100644 --- a/ge/ge_runtime/task/stream_active_task.cc +++ b/ge/ge_runtime/task/stream_active_task.cc @@ -27,6 +27,7 @@ StreamActiveTask::StreamActiveTask(const ModelContext &model_context, active_stream_(nullptr) { if (task_info_ == nullptr) { GELOGW("task_info_ is null!"); + return; } auto stream_list = model_context.stream_list(); uint32_t stream_id = task_info->stream_id(); @@ -34,6 +35,7 @@ StreamActiveTask::StreamActiveTask(const ModelContext &model_context, GELOGI("Stream list size:%zu, stream id:%u, active stream id:%u", stream_list.size(), stream_id, active_stream_id); if (stream_id >= stream_list.size() || active_stream_id >= stream_list.size()) { GELOGW("Stream id invalid"); + return; } stream_ = stream_list[stream_id]; active_stream_ = stream_list[active_stream_id]; diff --git a/ge/ge_runtime/task/stream_switch_task.cc b/ge/ge_runtime/task/stream_switch_task.cc index 91141139..afc5d32a 100644 --- a/ge/ge_runtime/task/stream_switch_task.cc +++ b/ge/ge_runtime/task/stream_switch_task.cc @@ -27,6 +27,7 @@ StreamSwitchTask::StreamSwitchTask(const ModelContext &model_context, stream_list_() { if (task_info_ == nullptr) { GELOGW("task_info_ is null!"); + return; } stream_list_ = model_context.stream_list(); diff --git a/ge/ge_runtime/task/tbe_task.cc b/ge/ge_runtime/task/tbe_task.cc index e7025ae8..07a864de 100644 --- a/ge/ge_runtime/task/tbe_task.cc +++ b/ge/ge_runtime/task/tbe_task.cc @@ -28,6 +28,7 @@ TbeTask::TbeTask(const ModelContext &model_context, const std::shared_ptr