PyTorch Forums
How can I get the group’s cuda stream,when backend is nccl
distributed
Amicoyuan
(Amicoyuan)
April 17, 2025, 4:32am
1
maybe like this: comm_stream = group.get_cuda_stream() ?