BaguaSys / bagua

Bagua Speeds up PyTorch

Home Page:https://tutorials-8ro.pages.dev/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

combine **inplace API

todo opened this issue · comments

commented

# TODO combine **inplace API
def alltoall_inplace(
tensor: torch.Tensor,
comm: Optional[B.BaguaSingleCommunicatorPy] = None,
):
"""The in-place version of :func:`alltoall`."""


This issue was generated by todo based on a TODO comment in cd499b8. It's been assigned to @wangraying because they committed the code.