Searched refs:set_wait_stream (Results 1 – 8 of 8) sorted by relevance
29 virtual void set_wait_stream(void *stream) = 0;
184 pre_event_->set_wait_stream(stream_); in Init()186 post_event_->set_wait_stream(compute_stream_); in Init()
34 …void set_wait_stream(void *wait_stream) override { wait_stream_ = static_cast<cudaStream_t>(wait_s… in set_wait_stream() function
34 void set_wait_stream(rtStream_t wait_stream) override { wait_stream_ = wait_stream; } in set_wait_stream() function
203 pre_event_->set_wait_stream(stream_); in Init()205 post_event_->set_wait_stream(compute_stream_); in Init()
826 event->set_wait_stream(wait_stream); in GenKernelEvents()835 pre_event->set_wait_stream(wait_stream); in GenKernelEvents()876 post_event->set_wait_stream(stream_); in ProcessBoundaryEvent()
712 pre_event->set_wait_stream(communication_stream_); in GenKernelEvents()714 post_event->set_wait_stream(stream_); in GenKernelEvents()
519 memcpy_event->set_wait_stream(model_stream); in LoadInputData()