Home
last modified time | relevance | path

Searched refs:task_map_ (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_parallel_build.cc141 task_map_[task_id] = task_info; in SaveTaskInfo()
145 MS_LOG(INFO) << "wait process task_num: " << task_map_.size(); in IsAllTaskFinish()
146 return task_map_.empty(); in IsAllTaskFinish()
171 auto task_iter = task_map_.find(task_id); in TaskFinishProcess()
172 if (task_iter == task_map_.end()) { in TaskFinishProcess()
185 (void)task_map_.erase(task_iter); in TaskFinishProcess()
213 (void)task_map_.erase(task_iter); in TaskFinishProcess()
214 MS_LOG(INFO) << "wait process remain task_num:" << task_map_.size(); in TaskFinishProcess()
314 task_map_.clear(); in ResetTaskInfo()
320 auto find_iter = task_map_.find(task_id); in GetAnfNodeByTaskID()
[all …]
Dtbe_kernel_parallel_build.h75 std::map<int32_t, KernelBuildTaskInfo> task_map_; variable
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc481 auto insert_ret = task_map_.insert(std::make_pair(graph.graph_id(), task_info_list)); in GenTask()
1139 auto iter = task_map_.find(graph.graph_id()); in GraphWithEmptyTaskList()
1140 if (iter == task_map_.end()) { in GraphWithEmptyTaskList()
1147 …return task_map_.find(graph_id) != task_map_.end() && graph_model_map_.find(graph_id) != graph_mod… in CheckGraphIdValid()
Dascend_kernel_runtime.h113 unordered_map<GraphId, vector<std::shared_ptr<TaskInfo>>> task_map_; variable