Mark DeviceMesh.__init__ as skip by angelayi · Pull Request #175510 · pytorch/pytorch

angelayi · 2026-02-23T02:55:17Z

Unsure whats the best way to fix things but this seems to work!

cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @kadeng @chauhang @amjames @Lucaskabela @jataylo

pytorch-bot · 2026-02-23T02:55:21Z

🔗 Helpful Links

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

📄 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.

✅ You can merge normally! (3 Unrelated Failures)

As of commit f6e27f2 with merge base 36bc2bf ():

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

pull / linux-jammy-py3.10-clang15 / test (dynamo_wrapped, 3, 3, linux.2xlarge) (gh) (disabled by #137892 but the issue was closed recently and a rebase is needed to make it pass)
test/test_autograd.py::TestAutograd::test_post_accumulate_grad_hook_e2e
trunk / linux-jammy-rocm-py3.10 / test (default, 1, 6, linux.rocm.gpu.gfx950.1) (gh) (disabled by #107893 but the issue was closed recently and a rebase is needed to make it pass)
test/inductor/test_inductor_freezing.py::FreezingCpuTests::test_conv_with_as_strided_cpu

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

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

trunk / linux-jammy-rocm-py3.10 / test (default, 4, 6, linux.rocm.gpu.gfx950.1) (gh) (trunk failure)
test/test_binary_ufuncs.py::TestBinaryUfuncsCUDA::test_fmod_remainder_by_zero_integral_cuda_uint8

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

pytorch-bot · 2026-02-23T02:55:23Z

This PR needs a `release notes:` label

If your changes are user facing and intended to be a part of release notes, please use a label starting with release notes:.

If not, please add the topic: not user facing label.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "topic: not user facing"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

azahed98

LGTM but I think the PR name should change to DeviceMesh instead of DTensor

Funnily enough I'm working on a related PR but for init on tensor subclasses #175397. What was the original error?

angelayi · 2026-02-25T20:54:48Z

The repro is in the test case, but the original error was, 'DeviceMesh' object has no attribute '_mesh_dim_names'

angelayi · 2026-02-25T20:54:53Z

@pytorchbot merge

pytorchmergebot · 2026-02-25T20:57:14Z

Merge failed

Reason: This PR needs a release notes: label
If your changes are user facing and intended to be a part of release notes, please use a label starting with release notes:.

If not, please add the topic: not user facing label.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "topic: not user facing"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

Details for Dev Infra team

Raised by workflow job

pytorch-bot · 2026-02-25T20:57:19Z

This PR needs a `release notes:` label

If your changes are user facing and intended to be a part of release notes, please use a label starting with release notes:.

If not, please add the topic: not user facing label.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "topic: not user facing"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

angelayi · 2026-02-25T20:58:10Z

@pytorchbot merge

pytorchmergebot · 2026-02-25T21:00:30Z

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging

Check the merge workflow status
here

malfet · 2026-02-26T03:30:01Z

@pytorchbot revert -m "It broke init_device_mesh_init_skip_after_graphbreak (whatever that means), see https://hud.pytorch.org/hud/pytorch/pytorch/d63f6584d251be2c9b9532f462e09399d9a29f03/1?per_page=50&name_filter=asan&mergeEphemeralLF=true for example" -c nosignal

pytorchmergebot · 2026-02-26T03:32:38Z

@pytorchbot successfully started a revert job. Check the current status here.
Questions? Feedback? Please reach out to the PyTorch DevX Team

This reverts commit 3918989. Reverted #175510 on behalf of https://github.com/malfet due to It broke init_device_mesh_init_skip_after_graphbreak (whatever that means), see https://hud.pytorch.org/hud/pytorch/pytorch/d63f6584d251be2c9b9532f462e09399d9a29f03/1?per_page=50&name_filter=asan&mergeEphemeralLF=true for example ([comment](#175510 (comment)))

pytorchmergebot · 2026-02-26T03:32:44Z

@angelayi your PR has been successfully reverted.

jeanschmidt · 2026-02-27T23:31:37Z

@pytorchbot revert -m "Need to revert in order to revert #175555 - see D94598146" -c ghfirst

pytorchmergebot · 2026-02-27T23:33:29Z

@pytorchbot successfully started a revert job. Check the current status here.
Questions? Feedback? Please reach out to the PyTorch DevX Team

This reverts commit 45c625b. Reverted #175510 on behalf of https://github.com/jeanschmidt due to Need to revert in order to revert #175555 - see D94598146 ([comment](#175510 (comment)))

pytorchmergebot · 2026-02-27T23:33:35Z

@angelayi your PR has been successfully reverted.

angelayi · 2026-03-02T16:34:20Z

@pytorchbot merge

pytorchmergebot · 2026-03-02T16:36:46Z

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging

Check the merge workflow status
here

Unsure whats the best way to fix things but this seems to work! Pull Request resolved: pytorch#175510 Approved by: https://github.com/azahed98

huydhn · 2026-03-04T20:16:08Z

@pytorchbot revert -m 'Per our discussion with @angelayi, revert this so that #169867 can be reverted, it is breaking a bunch of internal tests' -c ghfirst

pytorchmergebot · 2026-03-04T20:18:05Z

@pytorchbot successfully started a revert job. Check the current status here.
Questions? Feedback? Please reach out to the PyTorch DevX Team

@angelayi

This reverts commit 4416f11. Reverted #175510 on behalf of https://github.com/huydhn due to Per our discussion with @angelayi, revert this so that #169867 can be reverted, it is breaking a bunch of internal tests ([comment](#175510 (comment)))

pytorchmergebot · 2026-03-04T20:18:13Z

@angelayi your PR has been successfully reverted.

@angelayi

This reverts commit 4416f11. Reverted pytorch#175510 on behalf of https://github.com/huydhn due to Per our discussion with @angelayi, revert this so that pytorch#169867 can be reverted, it is breaking a bunch of internal tests ([comment](pytorch#175510 (comment)))

angelayi · 2026-03-11T15:54:19Z

relanded with #176661

Unsure whats the best way to fix things but this seems to work! Pull Request resolved: pytorch#175510 Approved by: https://github.com/azahed98

This reverts commit 3918989. Reverted pytorch#175510 on behalf of https://github.com/malfet due to It broke init_device_mesh_init_skip_after_graphbreak (whatever that means), see https://hud.pytorch.org/hud/pytorch/pytorch/d63f6584d251be2c9b9532f462e09399d9a29f03/1?per_page=50&name_filter=asan&mergeEphemeralLF=true for example ([comment](pytorch#175510 (comment)))

Unsure whats the best way to fix things but this seems to work! Pull Request resolved: pytorch#175510 Approved by: https://github.com/azahed98

This reverts commit 45c625b. Reverted pytorch#175510 on behalf of https://github.com/jeanschmidt due to Need to revert in order to revert pytorch#175555 - see D94598146 ([comment](pytorch#175510 (comment)))

Unsure whats the best way to fix things but this seems to work! Pull Request resolved: pytorch#175510 Approved by: https://github.com/azahed98

@angelayi

This reverts commit 4416f11. Reverted pytorch#175510 on behalf of https://github.com/huydhn due to Per our discussion with @angelayi, revert this so that pytorch#169867 can be reverted, it is breaking a bunch of internal tests ([comment](pytorch#175510 (comment)))

pytorch-bot Bot added ciflow/inductor module: dynamo labels Feb 23, 2026

angelayi force-pushed the angelayi/dtensor_init branch from 4d73f1d to cf5997a Compare February 23, 2026 17:13

angelayi marked this pull request as ready for review February 24, 2026 23:10

angelayi requested a review from williamwen42 February 24, 2026 23:11

angelayi changed the title ~~test dtensor init~~ Mark DTensor.__init__ as skip Feb 24, 2026

azahed98 approved these changes Feb 25, 2026

View reviewed changes

angelayi changed the title ~~Mark DTensor.__init__ as skip~~ Mark DeviceMesh.__init__ as skip Feb 25, 2026

pytorch-bot Bot added the ciflow/trunk Trigger trunk jobs on your pull request label Feb 25, 2026

pytorchmergebot added the merging label Feb 25, 2026

pytorchmergebot removed the merging label Feb 25, 2026

angelayi added the topic: not user facing topic category label Feb 25, 2026

pytorchmergebot added the merging label Feb 25, 2026

pytorchmergebot added the Merged label Feb 26, 2026

pytorchmergebot closed this in 3918989 Feb 26, 2026

pytorchmergebot removed the merging label Feb 26, 2026

github-actions Bot mentioned this pull request Feb 26, 2026

[Digest] 2026-02-26 e1134171019/agrnt#78

Open

pytorchmergebot closed this in 45c625b Feb 26, 2026

pytorchmergebot removed the merging label Feb 26, 2026

pytorchmergebot reopened this Feb 27, 2026

pytorchmergebot added the merging label Mar 2, 2026

pytorchmergebot closed this in 4416f11 Mar 2, 2026

pytorchmergebot removed the merging label Mar 2, 2026

postmath pushed a commit to postmath/pytorch that referenced this pull request Mar 3, 2026

Mark DeviceMesh.__init__ as skip (pytorch#175510)

dad1aae

Unsure whats the best way to fix things but this seems to work! Pull Request resolved: pytorch#175510 Approved by: https://github.com/azahed98

pytorchmergebot reopened this Mar 4, 2026

angelayi closed this Mar 11, 2026

github-actions Bot deleted the angelayi/dtensor_init branch April 11, 2026 02:23

Conversation

angelayi commented Feb 23, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

pytorch-bot Bot commented Feb 23, 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/175510

✅ You can merge normally! (3 Unrelated Failures)

Uh oh!

pytorch-bot Bot commented Feb 23, 2026

This PR needs a release notes: label

Uh oh!

azahed98 left a comment • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

angelayi commented Feb 25, 2026

Uh oh!

angelayi commented Feb 25, 2026

Uh oh!

pytorchmergebot commented Feb 25, 2026

Merge failed

Uh oh!

pytorch-bot Bot commented Feb 25, 2026

This PR needs a release notes: label

Uh oh!

angelayi commented Feb 25, 2026

Uh oh!

pytorchmergebot commented Feb 25, 2026

Merge started

Uh oh!

malfet commented Feb 26, 2026

Uh oh!

pytorchmergebot commented Feb 26, 2026

Uh oh!

pytorchmergebot commented Feb 26, 2026

Uh oh!

jeanschmidt commented Feb 27, 2026

Uh oh!

pytorchmergebot commented Feb 27, 2026

Uh oh!

pytorchmergebot commented Feb 27, 2026

Uh oh!

angelayi commented Mar 2, 2026

Uh oh!

pytorchmergebot commented Mar 2, 2026

Merge started

Uh oh!

huydhn commented Mar 4, 2026

Uh oh!

pytorchmergebot commented Mar 4, 2026

Uh oh!

pytorchmergebot commented Mar 4, 2026

Uh oh!

angelayi commented Mar 11, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

6 participants

angelayi commented Feb 23, 2026 •

edited

Loading

pytorch-bot Bot commented Feb 23, 2026 •

edited

Loading

This PR needs a `release notes:` label

azahed98 left a comment •

edited

Loading

This PR needs a `release notes:` label