Function c10::cuda::getStreamFromExternal¶
Defined in File CUDAStream.h
Function Documentation¶
-
CUDAStream c10::cuda::getStreamFromExternal(cudaStream_t ext_stream, DeviceIndex device_index)¶
Get a CUDAStream from a externally allocated one.
This is mainly for interoperability with different libraries where we want to operate on a non-torch allocated stream for data exchange or similar purposes