frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[BE][Easy] enable UFMT for torch/distributed/ (#128870)"

Browse Source 
This reverts commit a0e1e20c4157bb3e537fc784a51d7aef1e754157.

Reverted https://github.com/pytorch/pytorch/pull/128870 on behalf of https://github.com/fbgheith due to breaking internal builds ([comment](https://github.com/pytorch/pytorch/pull/128870#issuecomment-2181780356))
This commit is contained in:
PyTorch MergeBot
2024-06-21 00:38:28 +00:00
parent 9dd8f8cf8b
commit 9c929f6ce9
36 changed files with 299 additions and 584 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/distributed/_state_dict_utils.py
View File

@ -22,7 +22,6 @@ import torch.distributed as dist
import torch.nn.functional as F
from torch.distributed._functional_collectives import AsyncCollectiveTensor
if dist.is_available() or TYPE_CHECKING:
from torch.distributed import distributed_c10d
from torch.distributed._shard.sharded_tensor import ShardedTensor