Home
last modified time | relevance | path

Searched refs:kernel_post_run_events (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc700 auto &kernel_post_run_events = kernel_events.second; in GenKernelEvents() local
702 kernel_post_run_events.resize(kernels.size()); in GenKernelEvents()
720 kernel_post_run_events[i].emplace_back([post_event]() { post_event->RecordEvent(); }); in GenKernelEvents()
742 kernel_post_run_events[i].emplace_back([post_event]() { post_event->WaitEvent(); }); in GenKernelEvents()
1484 std::vector<std::vector<std::function<void()>>> kernel_post_run_events; in LaunchKernelMod() local
1488 kernel_post_run_events = events_iter->second.second; in LaunchKernelMod()
1525 LaunchKernelEvent(kernel_post_run_events, i); in LaunchKernelMod()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc797 auto &kernel_post_run_events = kernel_events.second; in GenKernelEvents() local
799 kernel_post_run_events.resize(kernels.size()); in GenKernelEvents()
828 kernel_post_run_events[IntToSize(k)].emplace_back([event]() { event->RecordEvent(); }); in GenKernelEvents()
841 ProcessBoundaryEvent(kernels, &kernel_post_run_events, last_stream_nodes); in GenKernelEvents()