Files
mscclpp/python
aashaka 2101f5251e Allow MSCCL++ CommGroup to take PyTorch tensors in args (#255)
Obtain data_ptr and tensor_size accordingly for torch.Tensor

Co-authored-by: Binyang Li <binyli@microsoft.com>
2024-02-06 19:47:25 -08:00
..
2024-02-04 20:46:10 -08:00
2024-02-04 20:46:10 -08:00