Method WaitForEvent

tensorflow/stream_executor/host/host_gpu_executor.cc:234–240

}

port::Status HostExecutor::WaitForEvent(Stream *stream, Event *event) {
  std::shared_ptr<absl::Notification> notification =
      AsHostEvent(event)->notification();
  AsHostStream(stream)->EnqueueTask(
      [notification]() { notification->WaitForNotification(); });
  return port::Status::OK();
}

Event::Status HostExecutor::PollForEventStatus(Event *event) {
  absl::Notification &notification = *AsHostEvent(event)->notification();