[DTensor] single-dim pointwise strategy#168115
[DTensor] single-dim pointwise strategy#168115wconstab wants to merge 23 commits intogh/wconstab/459/basefrom
Conversation
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/168115
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (2 Unrelated Failures)As of commit 6e1a7ad with merge base 74e112c ( FLAKY - The following job failed but was likely due to flakiness present on trunk:
BROKEN TRUNK - The following job failed but was present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
[ghstack-poisoned]
[ghstack-poisoned]
| return single_mesh_dim_common_pointwise_strategy(op_schema.args_schema, linearity) | ||
|
|
||
| def single_mesh_dim_common_pointwise_strategy( | ||
| args_schema: Sequence[object], |
There was a problem hiding this comment.
should contain tensormeta not Opstrategy/DTensorSpec
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
ghstack-source-id: ecded22 Pull Request resolved: pytorch/pytorch#168115
ghstack-source-id: dd60d43 Pull Request resolved: pytorch/pytorch#168115
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
| pointwise_strategy | ||
| ) | ||
|
|
||
| # register_single_dim_strategy( |
There was a problem hiding this comment.
purposefully not enabled yet. but I left the commented code here since I often want to switch it back on for testing
|
@pytorchbot merge -i |
Merge startedYour change will be merged while ignoring the following 1 checks: inductor / unit-test / inductor-pallas-cpu-test / test (inductor-pallas-cpu, 1, 1, linux.12xlarge) Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
This PR adds a dummy version of a single-dim pointwise strategy, and it shouldn't be used yet for real but it is being used by unit tests in #170631 so it is useful to land. Pull Request resolved: #168115 Approved by: https://github.com/pianpwk Co-authored-by: Pian Pawakapan <pianpwk@meta.com>
This PR adds a dummy version of a single-dim pointwise strategy, and it shouldn't be used yet for real but it is being used by unit tests in pytorch#170631 so it is useful to land. Pull Request resolved: pytorch#168115 Approved by: https://github.com/pianpwk Co-authored-by: Pian Pawakapan <pianpwk@meta.com>
Stack from ghstack (oldest at bottom):
This PR adds a dummy version of a single-dim pointwise strategy, and it shouldn't be used yet for real but it is being used by unit tests in #170631 so it is useful to land.