mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
94dc3253a0
Part of #123062 - #123062 Pull Request resolved: https://github.com/pytorch/pytorch/pull/128870 Approved by: https://github.com/fegin, https://github.com/wconstab
8 lines
145 B
Python
8 lines
145 B
Python
import torch
|
|
|
|
from .functional import * # noqa: F403
|
|
|
|
|
|
if torch.distributed.rpc.is_available():
|
|
from .api.remote_module import RemoteModule
|