Function c10::cuda::getDefaultCUDAStream¶
Defined in File CUDAStream.h
Function Documentation¶
-
CUDAStream c10::cuda::getDefaultCUDAStream(DeviceIndex device_index = -1)¶
Get the default CUDA stream, for the passed CUDA device, or for the current device if no device index is passed.
The default stream is where most computation occurs when you aren’t explicitly using streams.