mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-04-13 03:35:23 +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.