pytorch/distributed.tensor.parallel.rst at 045d1de02de46a972f8eab8196838ad73957a295 - pytorch - Carlos Sousa's Git

OSSForks/pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01: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

8 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