[DTensor] report registered ops by wconstab · Pull Request #176034 · pytorch/pytorch

wconstab · 2026-02-28T00:17:40Z

Stack from ghstack (oldest at bottom):

python -m torch.distributed.tensor._ops.strategy_validation --report

======================================================================
DTensor operator registration report
======================================================================

Directly registered:
  rule (register_prop_rule):               4
  op_strategy (register_op_strategy):    612
  single_dim_strategy:                    13
  total:                                 625

Decomposition table (not directly registered): 669
  These ops have entries in torch._decomp.decomposition_table but no
  direct DTensor strategy. They may work at runtime via
  DecompShardingStrategy if all decomposed sub-ops are supported.
  Additional ops beyond this count may also be reachable via CIA
  (CompositeImplicitAutograd) decompositions.

with --report-full

rule (4):
  aten.convolution.default
  aten.convolution_backward.default
  aten.index.Tensor
  aten.index_select.default

op_strategy (612):
  aten.__ilshift__.Scalar
  aten.__ilshift__.Tensor
  aten.__irshift__.Scalar
  aten.__irshift__.Tensor
  ...(truncated for git commit msg)

single_dim_strategy (13):
  aten._fft_c2c.default
  aten._fft_c2r.default
  aten._fft_r2c.default
  aten._index_put_impl_.default
  ...(truncated for git commit msg)

decomp table (not directly registered) (669):
  aten.__iand__.Scalar
  aten.__iand__.Tensor
  aten.__ior__.Scalar
  aten.__ior__.Tensor
  ...(truncated for git commit msg)

python -m torch.distributed.tensor._ops.strategy_validation --report ``` ====================================================================== DTensor operator registration report ====================================================================== Directly registered: rule (register_prop_rule): 4 op_strategy (register_op_strategy): 612 single_dim_strategy: 13 total: 625 Decomposition table (not directly registered): 669 These ops have entries in torch._decomp.decomposition_table but no direct DTensor strategy. They may work at runtime via DecompShardingStrategy if all decomposed sub-ops are supported. Additional ops beyond this count may also be reachable via CIA (CompositeImplicitAutograd) decompositions. ``` with --report-full ``` rule (4): aten.convolution.default aten.convolution_backward.default aten.index.Tensor aten.index_select.default op_strategy (612): aten.__ilshift__.Scalar aten.__ilshift__.Tensor aten.__irshift__.Scalar aten.__irshift__.Tensor ...(truncated for git commit msg) single_dim_strategy (13): aten._fft_c2c.default aten._fft_c2r.default aten._fft_r2c.default aten._index_put_impl_.default ...(truncated for git commit msg) decomp table (not directly registered) (669): aten.__iand__.Scalar aten.__iand__.Tensor aten.__ior__.Scalar aten.__ior__.Tensor ...(truncated for git commit msg) ``` [ghstack-poisoned]

pytorch-bot · 2026-02-28T00:17:44Z

🔗 Helpful Links

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

📄 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! (1 Unrelated Failure)

As of commit 4417437 with merge base 7eeab8a ():

FLAKY - The following job failed but was likely due to flakiness present on trunk:

trunk / macos-py3-arm64 / test (default, 2, 3, macos-m1-stable) (gh) (disabled by #176123 but the issue was closed recently and a rebase is needed to make it pass)
test/test_indexing.py::TestIndexingMPS::test_index_reduce_reduce_mean_mps_float32

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

python -m torch.distributed.tensor._ops.strategy_validation --report ``` ====================================================================== DTensor operator registration report ====================================================================== Directly registered: rule (register_prop_rule): 4 op_strategy (register_op_strategy): 612 single_dim_strategy: 13 total: 625 Decomposition table (not directly registered): 669 These ops have entries in torch._decomp.decomposition_table but no direct DTensor strategy. They may work at runtime via DecompShardingStrategy if all decomposed sub-ops are supported. Additional ops beyond this count may also be reachable via CIA (CompositeImplicitAutograd) decompositions. ``` with --report-full ``` rule (4): aten.convolution.default aten.convolution_backward.default aten.index.Tensor aten.index_select.default op_strategy (612): aten.__ilshift__.Scalar aten.__ilshift__.Tensor aten.__irshift__.Scalar aten.__irshift__.Tensor ...(truncated for git commit msg) single_dim_strategy (13): aten._fft_c2c.default aten._fft_c2r.default aten._fft_r2c.default aten._index_put_impl_.default ...(truncated for git commit msg) decomp table (not directly registered) (669): aten.__iand__.Scalar aten.__iand__.Tensor aten.__ior__.Scalar aten.__ior__.Tensor ...(truncated for git commit msg) ``` ghstack-source-id: 1fe849c Pull Request resolved: #176034

wconstab · 2026-03-02T22:02:08Z

@pytorchbot merge

pytorchmergebot · 2026-03-02T22:04:25Z

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

pytorchmergebot · 2026-03-03T04:02:35Z

The merge job was canceled or timed out. This most often happen if two merge requests were issued for the same PR, or if merge job was waiting for more than 6 hours for tests to finish. In later case, please do not hesitate to reissue the merge command
For more information see pytorch-bot wiki.

wconstab · 2026-03-03T04:49:30Z

@pytorchbot merge

wconstab · 2026-03-04T19:19:39Z

@pytorchbot merge

pytorchmergebot · 2026-03-04T19:22:35Z

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

python -m torch.distributed.tensor._ops.strategy_validation --report ``` ====================================================================== DTensor operator registration report ====================================================================== Directly registered: rule (register_prop_rule): 4 op_strategy (register_op_strategy): 612 single_dim_strategy: 13 total: 625 Decomposition table (not directly registered): 669 These ops have entries in torch._decomp.decomposition_table but no direct DTensor strategy. They may work at runtime via DecompShardingStrategy if all decomposed sub-ops are supported. Additional ops beyond this count may also be reachable via CIA (CompositeImplicitAutograd) decompositions. ``` with --report-full ``` rule (4): aten.convolution.default aten.convolution_backward.default aten.index.Tensor aten.index_select.default op_strategy (612): aten.__ilshift__.Scalar aten.__ilshift__.Tensor aten.__irshift__.Scalar aten.__irshift__.Tensor ...(truncated for git commit msg) single_dim_strategy (13): aten._fft_c2c.default aten._fft_c2r.default aten._fft_r2c.default aten._index_put_impl_.default ...(truncated for git commit msg) decomp table (not directly registered) (669): aten.__iand__.Scalar aten.__iand__.Tensor aten.__ior__.Scalar aten.__ior__.Tensor ...(truncated for git commit msg) ``` Pull Request resolved: pytorch#176034 Approved by: https://github.com/pianpwk ghstack dependencies: pytorch#175821

wconstab mentioned this pull request Feb 28, 2026

[DTensor] Strategy Validation fix for non-tensor args #175821

Closed

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

pianpwk approved these changes Mar 2, 2026

View reviewed changes

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

pytorchmergebot added the merging label Mar 2, 2026

pytorchmergebot added the Merged label Mar 4, 2026

pytorchmergebot closed this in 40bc62c Mar 4, 2026

pytorchmergebot removed the merging label Mar 4, 2026

github-actions Bot deleted the gh/wconstab/552/head branch April 4, 2026 02:23

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

[DTensor] report registered ops#176034

[DTensor] report registered ops#176034
wconstab wants to merge 1 commit intogh/wconstab/552/basefrom
gh/wconstab/552/head

wconstab commented Feb 28, 2026 •

edited

Loading

Uh oh!

pytorch-bot Bot commented Feb 28, 2026 •

edited

Loading

Uh oh!

wconstab commented Mar 2, 2026

Uh oh!

pytorchmergebot commented Mar 2, 2026

Uh oh!

pytorchmergebot commented Mar 3, 2026

Uh oh!

wconstab commented Mar 3, 2026

Uh oh!

wconstab commented Mar 4, 2026

Uh oh!

pytorchmergebot commented Mar 4, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

Conversation

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

Uh oh!

Uh oh!

pytorch-bot Bot commented Feb 28, 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/176034

✅ You can merge normally! (1 Unrelated Failure)

Uh oh!

wconstab commented Mar 2, 2026

Uh oh!

pytorchmergebot commented Mar 2, 2026

Merge started

Uh oh!

pytorchmergebot commented Mar 3, 2026

Uh oh!

wconstab commented Mar 3, 2026

Uh oh!

wconstab commented Mar 4, 2026

Uh oh!

pytorchmergebot commented Mar 4, 2026

Merge started

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

wconstab commented Feb 28, 2026 •

edited

Loading

pytorch-bot Bot commented Feb 28, 2026 •

edited

Loading