mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 17:26:04 +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.