mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-07-18 09:47:34 +00:00
Use `cuda_stream` attribute of a torch stream if the stream is not an instance of the cupy stream.
Use `cuda_stream` attribute of a torch stream if the stream is not an instance of the cupy stream.