Home
last modified time | relevance | path

Searched defs:WaitEvent (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_event.cc50 void GpuEvent::WaitEvent() { in WaitEvent() function in mindspore::device::gpu::GpuEvent
57 bool GpuEvent::WaitEvent(uint32_t stream_id) { in WaitEvent() function in mindspore::device::gpu::GpuEvent
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dascend_event.cc87 void AscendEvent::WaitEvent() { in WaitEvent() function in mindspore::device::ascend::AscendEvent
101 bool AscendEvent::WaitEvent(uint32_t stream_id) { in WaitEvent() function in mindspore::device::ascend::AscendEvent
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dmemory_manager.h93 bool WaitEvent(int64_t task_id_on_stream, uint32_t user_stream_id, uint32_t memory_stream_id) { in WaitEvent() function
100 bool WaitEvent(int64_t task_id_on_stream, uint32_t memory_stream_id) { in WaitEvent() function
Dmulti_stream_controller.cc107 bool MultiStreamController::WaitEvent(const DeviceContext *device_context, int64_t task_id_on_strea… in WaitEvent() function in mindspore::device::MultiStreamController
122 bool MultiStreamController::WaitEvent(const DeviceContext *device_context, int64_t task_id_on_strea… in WaitEvent() function in mindspore::device::MultiStreamController
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ccsrc/backend/common/
Dmem_dynamic_allocator_test.cc245 void WaitEvent() override{}; in WaitEvent() function in mindspore::device::DummyDeviceEvent
246 bool WaitEvent(uint32_t stream_id) override { return true; }; in WaitEvent() function in mindspore::device::DummyDeviceEvent
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/
Dmem_dynamic_allocator.cc1120 bool DynamicMemPoolBestFit::WaitEvent(int64_t task_id_on_stream, uint32_t user_stream_id, uint32_t … in WaitEvent() function in mindspore::device::DynamicMemPoolBestFit
1149 bool DynamicMemPoolBestFit::WaitEvent(int64_t task_id_on_stream, uint32_t memory_stream_id) { in WaitEvent() function in mindspore::device::DynamicMemPoolBestFit
1317 bool DynamicMemBuf::WaitEvent(uint32_t task_id_on_stream, uint32_t user_stream_id) { in WaitEvent() function in mindspore::device::DynamicMemBuf