pytorch/comm.py at main - pytorch - Gitea: Git for Me

frozenleaves/pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00

Files

Xuehai Pan f3fce597e9 [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769 )

See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501. Most changes are auto-generated by linter.

You can review these PRs via:

```bash
git diff --ignore-all-space --ignore-blank-lines HEAD~1
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/129769
Approved by: https://github.com/ezyang

2024-08-04 10:24:09 +00:00

20 lines

344 B

Python

Raw Permalink Blame History

 # The functions here have been moved to torch.nn.parallel.comm
 from torch.nn.parallel.comm import (
     broadcast,
     broadcast_coalesced,
     gather,
     reduce_add,
     reduce_add_coalesced,
     scatter,
 )
 __all__ = [
     "broadcast",
     "broadcast_coalesced",
     "reduce_add",
     "reduce_add_coalesced",
     "scatter",
     "gather",
 ]