File size: 520 Bytes
9dd3461 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# Keep old package for BC purposes, this file should be removed once
# everything moves to the `torch.distributed._shard` package.
import sys
import torch
import warnings
from torch.distributed._shard.sharding_spec import * # noqa: F403
warnings.warn(
"torch.distributed._sharding_spec will be deprecated, use torch.distributed._shard.sharding_spec instead",
DeprecationWarning
)
import torch.distributed._shard.sharding_spec as _sharding_spec
sys.modules['torch.distributed._sharding_spec'] = _sharding_spec
|