Searched refs:task_info_list (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 34 …tor::GenTasks(const std::vector<CNodePtr> &anf_node_list, std::vector<TaskInfoPtr> *task_info_list, in GenTasks() argument
37 MS_EXCEPTION_IF_NULL(task_info_list); in GenTasks()
39 if (!LaunchAllKernel(anf_node_list, task_info_list, graph_id)) { in GenTasks()
145 std::vector<TaskInfoPtr> *task_info_list) { in LaunchKernel() argument
146 MS_EXCEPTION_IF_NULL(task_info_list); in LaunchKernel()
236 task_info_list->insert(task_info_list->end(), task_info_ptrs.begin(), task_info_ptrs.end()); in LaunchKernel()
255 std::vector<TaskInfoPtr> *task_info_list, uint32_t graph_id) { in LaunchAllKernel() argument
260 size_t old_size = task_info_list->size(); in LaunchAllKernel()
265 if (!LaunchKernel(anf_node_ptr, stream_id, task_info_list)) { in LaunchAllKernel()
269 for (size_t i = old_size; i < task_info_list->size(); ++i) { in LaunchAllKernel()
|
| D | task_generator.h | 60 …bool GenTasks(const std::vector<CNodePtr> &anf_node_list, std::vector<TaskInfoPtr> *task_info_list,
69 …Kernel(const CNodePtr &anf_node_ptr, uint32_t stream_id, std::vector<TaskInfoPtr> *task_info_list);
70 …unchAllKernel(const std::vector<CNodePtr> &anf_node_list, std::vector<TaskInfoPtr> *task_info_list,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | label_set.cc | 54 std::vector<TaskInfoPtr> task_info_list; in GenTask() local
57 task_info_list.emplace_back(task_info_ptr); in GenTask()
58 return task_info_list; in GenTask()
|
| D | label_goto.cc | 54 std::vector<TaskInfoPtr> task_info_list; in GenTask() local
58 task_info_list.emplace_back(task_info_ptr); in GenTask()
59 return task_info_list; in GenTask()
|
| D | stream_active.cc | 75 std::vector<TaskInfoPtr> task_info_list; in GenTask() local
80 task_info_list.emplace_back(task_info_ptr); in GenTask()
83 return task_info_list; in GenTask()
|
| D | label_switch.cc | 63 std::vector<TaskInfoPtr> task_info_list; in GenTask() local
71 task_info_list.emplace_back(task_info_ptr); in GenTask()
72 return task_info_list; in GenTask()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | davinci_model.h | 27 DavinciModel(const std::vector<std::shared_ptr<TaskInfo>> &task_info_list,
33 : task_info_list_(task_info_list), in task_info_list_() argument
|
| /third_party/mindspore/tests/ut/cpp/stub/tasksink/ |
| D | task_sink_stub.cc | 26 …enTasks(const std::vector<CNodePtr> &anf_node_list, std::vector<TaskInfoPtr> *const task_info_list, in GenTasks() argument
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.cc | 474 vector<std::shared_ptr<TaskInfo>> task_info_list; in GenTask() local
477 if (!task_generator.GenTasks(anf_node_list, &task_info_list, graph.graph_id())) { in GenTask()
481 auto insert_ret = task_map_.insert(std::make_pair(graph.graph_id(), task_info_list)); in GenTask()
486 if (task_info_list.empty()) { in GenTask()
503 task_info_list, wait_active_stream_list, force_copy_stream_list, 0, 0, 0, 0, 0, 0, in GenTask()
|