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

[BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/153225
Approved by: https://github.com/kwen2501, https://github.com/fegin
This commit is contained in:
Xilun Wu
2025-05-09 01:50:00 -07:00
committed by PyTorch MergeBot
parent 3976e52264
commit cbb03e6971
59 changed files with 182 additions and 179 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/distributed/tensor/test_pointwise_ops.py
View File

@ -8,8 +8,10 @@ from unittest import skip
import torch
import torch.utils._pytree as pytree
from torch import Tensor
from torch.distributed._tensor import DeviceMesh, distribute_tensor, DTensor
from torch.distributed._tensor.placement_types import (
from torch.distributed.tensor import (
DeviceMesh,
distribute_tensor,
DTensor,
Partial,
Placement,
Replicate,