Searched defs:set_wait_stream (Results 1 – 2 of 2) sorted by relevance
34 void set_wait_stream(rtStream_t wait_stream) override { wait_stream_ = wait_stream; } in set_wait_stream() function
34 …void set_wait_stream(void *wait_stream) override { wait_stream_ = static_cast<cudaStream_t>(wait_s… in set_wait_stream() function