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 "[dtensor] move DTensor to public namespace (#133113)"

Browse Source 
This reverts commit 2ee6b97464d17fcf4c1fc67c29868fa30d0c16e1.

Reverted https://github.com/pytorch/pytorch/pull/133113 on behalf of https://github.com/wanchaol due to looks like it break some internal type imports ([comment](https://github.com/pytorch/pytorch/pull/133113#issuecomment-2295670911))
This commit is contained in:
PyTorch MergeBot
2024-08-19 05:00:19 +00:00
parent 42e61c783c
commit 35f36363ec
88 changed files with 265 additions and 433 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/distributed/_tensor/test_dtensor.py
View File

@ -909,7 +909,7 @@ class TestDTensorPlacementTypes(DTensorTestBase):
]
assert_array_equal(expected_pad_sizes, pad_sizes)
from torch.distributed.tensor._collective_utils import unpad_tensor
from torch.distributed._tensor._collective_utils import unpad_tensor
unpadded_list = [
unpad_tensor(tensor, shard_placement.dim, pad_sizes[i])