Searched refs:wait_stream (Results 1 – 3 of 3) sorted by relevance
34 …void set_wait_stream(void *wait_stream) override { wait_stream_ = static_cast<cudaStream_t>(wait_s… in set_wait_stream() argument
34 void set_wait_stream(rtStream_t wait_stream) override { wait_stream_ = wait_stream; } in set_wait_stream() argument
806 auto wait_stream = stream_id_map_[curr_stream_id]; in GenKernelEvents() local826 event->set_wait_stream(wait_stream); in GenKernelEvents()833 if (!found_depend && wait_stream != stream_) { in GenKernelEvents()835 pre_event->set_wait_stream(wait_stream); in GenKernelEvents()