Skip to content

Commit 52a5e70

Browse files
committed
code clean up
1 parent 04c5cd3 commit 52a5e70

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

torch_xla/distributed/spmd/xla_sharded_tensor.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,7 @@
99
import torch_xla.runtime as xr
1010
from torch.distributed.tensor._dtensor_spec import DTensorSpec, TensorMeta
1111
from torch.distributed.device_mesh import DeviceMesh
12-
<<<<<<< HEAD
1312
from torch.distributed.tensor.placement_types import Placement, Shard, Replicate, Partial
14-
=======
15-
from torch.distributed.tensor.placement_types import Shard, Replicate
16-
>>>>>>> 566959e10 (Removed auto wrapping sharding propagation, added cached spec invalidation)
1713
from torch.utils._pytree import tree_map_only
1814
from torch.distributed.tensor import DTensor
1915

0 commit comments

Comments
 (0)