Searched refs:TaskInfoType (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 27 enum TaskInfoType { enum
51 TaskInfoType type() const { return type_; } in type()
56 TaskInfo(const std::string &op_name, uint32_t stream_id, TaskInfoType type, bool dump_flag) in TaskInfo()
62 TaskInfoType type_;
72 : TaskInfo(op_name, stream_id, TaskInfoType::TBE, dump_flag), in TbeTaskInfo()
123 : TaskInfo(op_name, stream_id, TaskInfoType::AICPU, dump_flag), in AicpuTaskInfo()
151 : TaskInfo(op_name, stream_id, TaskInfoType::LABEL_SET, false), label_id_(label_id) {} in LabelSetTaskInfo()
162 : TaskInfo(op_name, stream_id, TaskInfoType::LABEL_GOTO, false), label_id_(label_id) {} in LabelGotoTaskInfo()
174 : TaskInfo(op_name, stream_id, TaskInfoType::LABEL_SWITCH, false), in LabelSwitchTaskInfo()
194 …EventTaskInfo(const std::string &op_name, uint32_t stream_id, TaskInfoType type, uint32_t event_id) in EventTaskInfo()
[all …]
|
| D | runtime_model.cc | 111 if (task_info->type() != TaskInfoType::LABEL_SET) { in InitLabel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | task_factory.h | 36 void RegisterCreator(const TaskInfoType &type, const TASK_CREATOR_FUN &func) { in RegisterCreator()
43 std::map<TaskInfoType, TASK_CREATOR_FUN> creator_map_;
67 Register(const TaskInfoType &type, const TASK_CREATOR_FUN &func) { in Register()
|
| D | event_record_task.cc | 53 REGISTER_TASK(TaskInfoType::EVENT_RECORD, EventRecordTask, EventRecordTaskInfo);
|
| D | memcpy_async_task.cc | 50 REGISTER_TASK(TaskInfoType::MEMCPY_ASYNC, MemcpyAsyncTask, MemcpyAsyncTaskInfo);
|
| D | stream_active_task.cc | 55 REGISTER_TASK(TaskInfoType::STREAM_ACTIVE, StreamActiveTask, StreamActiveTaskInfo);
|
| D | event_wait_task.cc | 58 REGISTER_TASK(TaskInfoType::EVENT_WAIT, EventWaitTask, EventWaitTaskInfo);
|
| D | label_set_task.cc | 55 REGISTER_TASK(TaskInfoType::LABEL_SET, LabelSetTask, LabelSetTaskInfo);
|
| D | profiler_task.cc | 57 REGISTER_TASK(TaskInfoType::PROFILER_TRACE, ProfilerTask, ProfilerTraceTaskInfo);
|
| D | label_switch_task.cc | 76 REGISTER_TASK(TaskInfoType::LABEL_SWITCH, LabelSwitchTask, LabelSwitchTaskInfo);
|
| D | stream_switch_task.cc | 70 REGISTER_TASK(TaskInfoType::STREAM_SWITCH, StreamSwitchTask, StreamSwitchTaskInfo);
|
| D | label_goto_task.cc | 82 REGISTER_TASK(TaskInfoType::LABEL_GOTO, LabelGotoTask, LabelGotoTaskInfo);
|
| D | tbe_task.cc | 96 REGISTER_TASK(TaskInfoType::TBE, TbeTask, TbeTaskInfo);
|
| D | aicpu_task.cc | 168 REGISTER_TASK(TaskInfoType::AICPU, AicpuTask, AicpuTaskInfo);
|
| D | hccl_task.cc | 219 REGISTER_TASK(TaskInfoType::HCCL, HcclTask, HcclTaskInfo);
|