[DTensor] Complete single-dim pointwise rule by wconstab · Pull Request #172278 · pytorch/pytorch

wconstab · 2026-01-12T22:34:08Z

Stack from ghstack (oldest at bottom):

Claude..

Claude.. [ghstack-poisoned]

pytorch-bot · 2026-01-12T22:34:11Z

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/172278

📄 Preview Python docs built from this PR
📄 Preview C++ docs built from this PR
❓ Need help or want to give feedback on the CI? Visit the bot commands wiki

Note: Links to docs will display an error until the docs builds have been completed.

❌ 3 New Failures, 5 Unrelated Failures

As of commit a435782 with merge base a6cd1b0 ():

NEW FAILURES - The following jobs have failed:

Check mergeability of ghstack PR / ghstack-mergeability-check (gh)
RuntimeError: Command git -C /home/runner/work/pytorch/pytorch cherry-pick -x fb0b701 returned non-zero exit code 1
Lint / lintrunner-pyrefly-partial / linux-job (gh)
>>> Lint for torch/distributed/tensor/_ops/strategy_validation.py:
pull / linux-jammy-py3.10-gcc11 / test (distributed, 2, 2, linux.2xlarge) (gh)
test/distributed/tensor/test_pointwise_ops.py::PointwisePartialsTest::test_add_partial_avg_with_replicate

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

inductor / unit-test / inductor-halide-test / test (inductor-halide, 1, 1, linux.12xlarge) (gh) (disabled by #150624 but the issue was closed recently and a rebase is needed to make it pass)
test/inductor/test_halide.py::HalideCpuTests::test_special_polygamma_cpu_halide
pull / cuda12.8-py3.10-gcc11-sm75 / test (pr_time_benchmarks, 1, 1, linux.g4dn.metal.nvidia.gpu) (gh) (similar failure)
MISSING REGRESSION TEST
pull / cuda13.0-py3.10-gcc11-sm75 / test (pr_time_benchmarks, 1, 1, linux.g4dn.metal.nvidia.gpu) (gh) (similar failure)
MISSING REGRESSION TEST

BROKEN TRUNK - The following jobs failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

inductor / inductor-test / test (inductor_timm, 1, 2, linux.g5.4xlarge.nvidia.gpu) (gh) (trunk failure)
Process completed with exit code 1.
inductor / inductor-test-cuda13 / test (inductor_timm, 1, 2, linux.g5.4xlarge.nvidia.gpu) (gh) (trunk failure)
Process completed with exit code 1.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

Claude.. ghstack-source-id: c6bb1dc Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: a3b2eda Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: 347f659 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: 4de4aa7 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: 96988e6 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: 5c5b22b Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: a4620ee Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: cb1e3f9 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: 0ac910b Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: ada24e9 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: 5eaf34a Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: 0ca9783 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: b4ff144 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: e17d4b9 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: df0e8b0 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: 1bc1019 Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

Claude.. ghstack-source-id: 0df63ec Pull Request resolved: #172278

Claude.. [ghstack-poisoned]

wconstab · 2026-02-06T20:52:59Z

handing off to @anshul-si

[DTensor] Complete single-dim pointwise rule

e737758

Claude.. [ghstack-poisoned]

wconstab mentioned this pull request Jan 12, 2026

[DTensor] Handle out= ops in single-dim expander #172276

Closed

wconstab mentioned this pull request Jan 12, 2026

[DTensor] Fix partial redistribution order #172277

Closed

pytorch-bot Bot added ciflow/inductor release notes: distributed (dtensor) release notes category labels Jan 12, 2026

wconstab added a commit that referenced this pull request Jan 12, 2026

[DTensor] Complete single-dim pointwise rule

0101e2a

Claude.. ghstack-source-id: c6bb1dc Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

4666fde

Claude.. [ghstack-poisoned]

wconstab mentioned this pull request Jan 12, 2026

[DTensor] Make Replicate->Partial cost > 0 #172282

Closed

wconstab added a commit that referenced this pull request Jan 12, 2026

[DTensor] Complete single-dim pointwise rule

c7ca920

Claude.. ghstack-source-id: a3b2eda Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

08a83dd

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 12, 2026

[DTensor] Complete single-dim pointwise rule

93b7d0f

Claude.. ghstack-source-id: 347f659 Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

b6ade31

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 13, 2026

[DTensor] Complete single-dim pointwise rule

cac3521

Claude.. ghstack-source-id: 4de4aa7 Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

53e2f2f

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 13, 2026

[DTensor] Complete single-dim pointwise rule

dffd8b8

Claude.. ghstack-source-id: 96988e6 Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

046ee83

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 13, 2026

[DTensor] Complete single-dim pointwise rule

810fbcd

Claude.. ghstack-source-id: 5c5b22b Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

a060846

Claude.. [ghstack-poisoned]

This was referenced Jan 14, 2026

[DTensor] make expand_to_full_mesh_op_strategy filter incompatible out= strategies #172420

Closed

[DTensor] single_dim fix symint + _create_expanded_strategy #172421

Closed

wconstab added a commit that referenced this pull request Jan 14, 2026

[DTensor] Complete single-dim pointwise rule

9dd3739

Claude.. ghstack-source-id: a4620ee Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

2abaf28

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 14, 2026

[DTensor] Complete single-dim pointwise rule

7ba89ee

Claude.. ghstack-source-id: cb1e3f9 Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

12be01b

Claude.. [ghstack-poisoned]

This was referenced Jan 14, 2026

[DTensor] single dim fix inplace op expansion #172477

Closed

[DTensor] fix redistribute cost crashing on non-participating ranks #172478

Closed

[DTensor] Make RedistributionPlanner handle all partials #172479

Closed

wconstab added a commit that referenced this pull request Jan 14, 2026

[DTensor] Complete single-dim pointwise rule

30ed365

Claude.. ghstack-source-id: 0ac910b Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

7875fa3

Claude.. [ghstack-poisoned]

Update on "[DTensor] Complete single-dim pointwise rule"

ce35cba

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 27, 2026

[DTensor] Complete single-dim pointwise rule

8bf5b4d

Claude.. ghstack-source-id: ada24e9 Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

67989a9

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 27, 2026

[DTensor] Complete single-dim pointwise rule

d672261

Claude.. ghstack-source-id: 5eaf34a Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

8371edd

Claude.. [ghstack-poisoned]

wconstab mentioned this pull request Jan 27, 2026

[DTensor] Update TP api to support single-dim strategies #173567

Closed

wconstab added a commit that referenced this pull request Jan 27, 2026

[DTensor] Complete single-dim pointwise rule

1b2d1f8

Claude.. ghstack-source-id: 0ca9783 Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

c999b5e

Claude.. [ghstack-poisoned]

wconstab mentioned this pull request Jan 27, 2026

[DTensor] single-dim expander raises clear inplace error #173572

Closed

wconstab added a commit that referenced this pull request Jan 27, 2026

[DTensor] Complete single-dim pointwise rule

643be3b

Claude.. ghstack-source-id: b4ff144 Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

eb83133

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 27, 2026

[DTensor] Complete single-dim pointwise rule

393e886

Claude.. ghstack-source-id: e17d4b9 Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

02d29d6

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 28, 2026

[DTensor] Complete single-dim pointwise rule

71645b0

Claude.. ghstack-source-id: df0e8b0 Pull Request resolved: #172278

wconstab mentioned this pull request Jan 28, 2026

[DTensor] expand_to_full_mesh_op_strategy filters mixed partials #173614

Closed

Update on "[DTensor] Complete single-dim pointwise rule"

1781962

Claude.. [ghstack-poisoned]

wconstab added a commit that referenced this pull request Jan 29, 2026

[DTensor] Complete single-dim pointwise rule

77c7241

Claude.. ghstack-source-id: 1bc1019 Pull Request resolved: #172278

wconstab added a commit that referenced this pull request Jan 29, 2026

[DTensor] Complete single-dim pointwise rule

95142af

Claude.. ghstack-source-id: 1bc1019 Pull Request resolved: #172278

Update on "[DTensor] Complete single-dim pointwise rule"

40c9dc2

Claude.. [ghstack-poisoned]

This was referenced Jan 31, 2026

[DTensor] Test pointwise partial propagation #174000

Closed

[LocalTensor] Cache DeviceMesh.get_coordinate results in LocalTensorMode #173836

Closed

[DTensor] Strategy Validation #173976

Closed

[DTensor] Add support for torch.linalg operators #171935

Closed

wconstab added a commit that referenced this pull request Jan 31, 2026

[DTensor] Complete single-dim pointwise rule

2ff7920

Claude.. ghstack-source-id: 0df63ec Pull Request resolved: #172278

wconstab mentioned this pull request Feb 2, 2026

WIP claude rewrite pointwise #174080

Closed

wconstab added 2 commits February 2, 2026 10:02

Update on "[DTensor] Complete single-dim pointwise rule"

07acf3b

Claude.. [ghstack-poisoned]

Update on "[DTensor] Complete single-dim pointwise rule"

a435782

Claude.. [ghstack-poisoned]

wconstab closed this Feb 6, 2026

github-actions Bot deleted the gh/wconstab/494/head branch March 9, 2026 02:21

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

[DTensor] Complete single-dim pointwise rule#172278

[DTensor] Complete single-dim pointwise rule#172278
wconstab wants to merge 26 commits intogh/wconstab/494/basefrom
gh/wconstab/494/head

wconstab commented Jan 12, 2026 •

edited

Loading

Uh oh!

pytorch-bot Bot commented Jan 12, 2026 •

edited

Loading

Uh oh!

wconstab commented Feb 6, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

Conversation

wconstab commented Jan 12, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

pytorch-bot Bot commented Jan 12, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/172278

❌ 3 New Failures, 5 Unrelated Failures

Uh oh!

wconstab commented Feb 6, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

wconstab commented Jan 12, 2026 •

edited

Loading

pytorch-bot Bot commented Jan 12, 2026 •

edited

Loading