pytorch
4451eb24 - Move tensor_parallel out to distributed.tensor folder (#89878)

Commit
3 years ago
Move tensor_parallel out to distributed.tensor folder (#89878) This PR moves tensor parallel from torch.distributed._tensor.parallel to torch.distributed.tensor.parallel, to prepare for beta release Pull Request resolved: https://github.com/pytorch/pytorch/pull/89878 Approved by: https://github.com/fduwjj
Author
Committer
Parents
Loading