1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
# 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.sharded_tensor import * # noqa: F403
with warnings.catch_warnings():
warnings.simplefilter("always")
warnings.warn(
"`torch.distributed._sharded_tensor` will be deprecated, "
"use `torch.distributed._shard.sharded_tensor` instead",
DeprecationWarning,
stacklevel=2,
)
sys.modules[
"torch.distributed._sharded_tensor"
] = torch.distributed._shard.sharded_tensor
|
