1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
# Keep old package for BC purposes, this file should be removed once
# everything moves to the `torch.distributed._shard` package.
import sys
import warnings
import torch
from torch.distributed._shard.sharding_spec import * # noqa: F403
with warnings.catch_warnings():
warnings.simplefilter("always")
warnings.warn(
"`torch.distributed._sharding_spec` will be deprecated, "
"use `torch.distributed._shard.sharding_spec` instead",
DeprecationWarning,
stacklevel=2,
)
import torch.distributed._shard.sharding_spec as _sharding_spec
sys.modules["torch.distributed._sharding_spec"] = _sharding_spec
|
