10000 [dtensor] PART 6: move DTensor op tests to core distributed by wanchaol · Pull Request #88551 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

[dtensor] PART 6: move DTensor op tests to core distributed #88551

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 10 commits into from
Closed

[dtensor] PART 6: move DTensor op tests to core distributed #88551

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
2630cc3
[dtensor] PART 6: move DTensor op tests to core distributed
wanchaol Nov 6, 2022
1bfce6e
Update on "[dtensor] PART 6: move DTensor op tests to core distributed"
wanchaol Nov 6, 2022
170eb26
Update on "[dtensor] PART 6: move DTensor op tests to core distributed"
wanchaol Nov 6, 2022
a6bc864
Update on "[dtensor] PART 6: move DTensor op tests to core distributed"
wanchaol Nov 6, 2022
355e48a
Update on "[dtensor] PART 6: move DTensor op tests to core distributed"
wanchaol Nov 7, 2022
7032f36
Update on "[dtensor] PART 6: move DTensor op tests to core distributed"
wanchaol Nov 10, 2022
651cf8a
Update on "[dtensor] PART 6: move DTensor op tests to core distributed"
wanchaol Nov 10, 2022
bb9281f
Update on "[dtensor] PART 6: move DTensor op tests to core distributed"
wanchaol Nov 15, 2022
36e7335
Update on "[dtensor] PART 6: move DTensor op tests to core distributed"
wanchaol Nov 15, 2022
19eeadb
Update on "[dtensor] PART 6: move DTensor op tests to core distributed"
wanchaol Nov 15, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update on "[dtensor] PART 6: move DTensor op tests to core distributed" 
This PR moves DTensor op tests to core distributed, including
prop_rule, pointwise op, matrix op tests, etc.

part of #88838

[ghstack-poisoned]
  • Loading branch information
@wanchaol
wanchaol committed Nov 15, 2022
commit 36e73355de95a91cd6896666a4fe21bc82a90f0b
2 changes: 1 addition & 1 deletion test/distributed/_tensor/test_common_rules.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
pointwise_rule,
)
from torch.distributed._tensor.placement_types import DTensorSpec
from torch.testing._internal.common_dtensor import (
from torch.testing._internal.distributed._tensor.common_dtensor import (
DTensorTestBase,
with_comms,
)
Expand Down
2 changes: 1 addition & 1 deletion test/distributed/_tensor/test_math_ops.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

from torch.distributed._tensor import distribute_tensor
from torch.distributed._tensor.placement_types import Shard, Replicate
from torch.testing._internal.common_dtensor import (
from torch.testing._internal.distributed._tensor.common_dtensor import (
DTensorTestBase,
with_comms,
skip_unless_torch_gpu,
Expand Down
2 changes: 1 addition & 1 deletion test/distributed/_tensor/test_matrix_ops.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import torch
from torch.testing._internal.common_utils import run_tests
from torch.distributed._tensor.api import DTensor
from torch.testing._internal.common_dtensor import (
from torch.testing._internal.distributed._tensor.common_dtensor import (
DTensorTestBase,
with_comms,
skip_unless_torch_gpu,
Expand Down
2 changes: 1 addition & 1 deletion test/distributed/_tensor/test_pointwise_ops.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
import torch
from torch import Tensor
from torch.testing._internal.common_utils import run_tests
from torch.testing._internal.common_dtensor import (
from torch.testing._internal.distributed._tensor.common_dtensor import (
DTensorTestBase,
with_comms,
skip_unless_torch_gpu,
Expand Down
2 changes: 1 addition & 1 deletion test/distributed/_tensor/test_tp_sharding_ops.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

import torch
from torch.testing._internal.common_utils import run_tests
from torch.testing._internal.common_dtensor import (
from torch.testing._internal.distributed._tensor.common_dtensor import (
DTensorTestBase,
with_comms,
)
Expand Down
2 changes: 1 addition & 1 deletion test/distributed/_tensor/test_view_ops.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from typing import List, cast
from torch.distributed._tensor.placement_types import Placement
from torch.testing._internal.common_dtensor import (
from torch.testing._internal.distributed._tensor.common_dtensor import (
DTensorTestBase,
redistribute_profiler,
with_comms,
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.
0