Function c10::cuda::setCurrentCUDAStream

• Defined in File CUDAStream.h

Function Documentation

void c10::cuda::setCurrentCUDAStream(CUDAStream stream)

Set the current stream on the device of the passed in stream to be the passed in stream.

Yes, you read that right: this function has nothing to do with the current device: it toggles the current stream of the device of the passed stream.

Confused? Avoid using this function; prefer using ‘CUDAStreamGuard’ instead (which will switch both your current device and current stream in the way you expect, and reset it back to its original state afterwards).