Alias of `oneflow.comm.send()` for PyTorch compatibility. See also :func:`oneflow.comm.send()`
(tensor: flow.Tensor, dst: int, group=None, tag: int = 0)
| 64 | |
| 65 | |
| 66 | def send(tensor: flow.Tensor, dst: int, group=None, tag: int = 0) -> None: |
| 67 | """Alias of `oneflow.comm.send()` for PyTorch compatibility. |
| 68 | |
| 69 | See also :func:`oneflow.comm.send()` |
| 70 | """ |
| 71 | assert group is None, "group is not supported yet" |
| 72 | assert tag == 0, "tag is not supported yet" |
| 73 | return flow.comm.send(tensor, dst) |
| 74 | |
| 75 | |
| 76 | def recv(tensor: flow.Tensor, src: int, group=None, tag: int = 0) -> None: |