Home
last modified time | relevance | path

Searched refs:WaitEvent (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/core/ir/
Dtensor.h82 class WaitEvent : public ExceptionListener {
91 MsException::Instance().SetExceptionListener(const_cast<WaitEvent *>(this)); in Wait()
329 event_ = std::make_shared<WaitEvent>(); in SetNeedWait()
348 device_event_->WaitEvent(); in WaitDevice()
378 mutable std::shared_ptr<WaitEvent> event_{nullptr};
Ddevice_event.h24 virtual void WaitEvent() = 0;
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_event.h29 void WaitEvent() override;
Dgpu_event.cc31 void GpuEvent::WaitEvent() { in WaitEvent() function in mindspore::device::gpu::GpuEvent
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_event.h29 void WaitEvent() override;
Dascend_event.cc57 void AscendEvent::WaitEvent() { in WaitEvent() function in mindspore::device::ascend::AscendEvent
Dascend_kernel_runtime.cc829 kernel_pre_run_events[i].emplace_back([event]() { event->WaitEvent(); }); in GenKernelEvents()
838 kernel_pre_run_events[i].emplace_back([pre_event]() { pre_event->WaitEvent(); }); in GenKernelEvents()
879 (*kernel_run_events)[i].emplace_back([post_event]() { post_event->WaitEvent(); }); in ProcessBoundaryEvent()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dbucket.cc52 pre_event_->WaitEvent(); in Launch()
Dkernel_runtime.cc718 pre_event->WaitEvent(); in GenKernelEvents()
737 kernel_pre_run_events[j].emplace_back([post_event]() { post_event->WaitEvent(); }); in GenKernelEvents()
742 kernel_post_run_events[i].emplace_back([post_event]() { post_event->WaitEvent(); }); in GenKernelEvents()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_session.cc522 memcpy_event->WaitEvent(); in LoadInputData()