pytorch/docs/source/distributed.tensor.parallel.rst at 7fe3c64bdb6dc5aa969230ce0b10a9869849b49e - saymrwulf/pytorch - Forgejo: Beyond coding. We forge.

saymrwulf/pytorch

mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00

Wanchao Liang 4451eb24e6 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

2022-11-30 22:13:10 +00:00

7 lines

191 B

ReStructuredText

Raw Blame History

 .. role:: hidden
     :class: hidden-section
 Tensor Parallelism
 ========================
 .. py:module:: torch.distributed.tensor.parallel
 .. currentmodule:: torch.distributed.tensor.parallel