Searched refs:EventWaitTaskInfo (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | event_wait_task.h | 24 class EventWaitTask : public TaskRepeater<EventWaitTaskInfo> {
26 …EventWaitTask(const ModelContext &model_context, const std::shared_ptr<EventWaitTaskInfo> &task_in…
33 std::shared_ptr<EventWaitTaskInfo> task_info_;
|
| D | event_wait_task.cc | 22 …entWaitTask(const ModelContext &model_context, const std::shared_ptr<EventWaitTaskInfo> &task_info) in EventWaitTask()
23 : TaskRepeater<EventWaitTaskInfo>(model_context, task_info), in EventWaitTask()
58 REGISTER_TASK(TaskInfoType::EVENT_WAIT, EventWaitTask, EventWaitTaskInfo);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | recv.cc | 25 using mindspore::ge::model_runner::EventWaitTaskInfo;
26 using EventWaitTaskInfoPtr = std::shared_ptr<EventWaitTaskInfo>;
59 …EventWaitTaskInfoPtr task_info_ptr = std::make_shared<EventWaitTaskInfo>(unique_name_, stream_id, … in GenTask()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 208 class EventWaitTaskInfo : public EventTaskInfo {
210 EventWaitTaskInfo(const std::string &op_name, uint32_t stream_id, uint32_t event_id) in EventWaitTaskInfo() function
212 ~EventWaitTaskInfo() override {} in ~EventWaitTaskInfo()
|
| /third_party/mindspore/tests/ut/cpp/device/ |
| D | ge_runtime_test.cc | 127 …std::shared_ptr<TaskInfo> event_record_task_info = std::make_shared<EventWaitTaskInfo>("op_name", … in TEST_F()
137 …std::shared_ptr<TaskInfo> event_record_task_info = std::make_shared<EventWaitTaskInfo>("op_name", … in TEST_F()
|