Skip to content

[DTensor] single-dim pointwise strategy#168115

Closed
wconstab wants to merge 23 commits intogh/wconstab/459/basefrom
gh/wconstab/459/head
Closed

[DTensor] single-dim pointwise strategy#168115
wconstab wants to merge 23 commits intogh/wconstab/459/basefrom
gh/wconstab/459/head

Conversation

@wconstab
Copy link
Contributor

@wconstab wconstab commented Nov 18, 2025

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.

@pytorch-bot
Copy link

pytorch-bot bot commented Nov 18, 2025

🔗 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 (image):

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.

return single_mesh_dim_common_pointwise_strategy(op_schema.args_schema, linearity)

def single_mesh_dim_common_pointwise_strategy(
args_schema: Sequence[object],
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

should contain tensormeta not Opstrategy/DTensorSpec

tiendatngcs pushed a commit to tiendatngcs/pytorch-Dec25 that referenced this pull request Dec 10, 2025
tiendatngcs pushed a commit to tiendatngcs/pytorch-Dec25 that referenced this pull request Dec 10, 2025
wconstab added a commit that referenced this pull request Dec 17, 2025
ghstack-source-id: df6871a
Pull Request resolved: #168115
wconstab added a commit that referenced this pull request Dec 17, 2025
ghstack-source-id: 9457c1b
Pull Request resolved: #168115
wconstab added a commit that referenced this pull request Dec 18, 2025
ghstack-source-id: f62b740
Pull Request resolved: #168115
wconstab added a commit that referenced this pull request Dec 18, 2025
ghstack-source-id: 1b5cc2e
Pull Request resolved: #168115
[ghstack-poisoned]
weifengpy pushed a commit that referenced this pull request Dec 19, 2025
ghstack-source-id: 9457c1b
Pull Request resolved: #168115
weifengpy pushed a commit that referenced this pull request Dec 19, 2025
ghstack-source-id: 9457c1b
Pull Request resolved: #168115
wconstab added a commit that referenced this pull request Dec 19, 2025
ghstack-source-id: 01f48e8
Pull Request resolved: #168115
@wconstab wconstab added the topic: not user facing topic category label Dec 19, 2025
pointwise_strategy
)

# register_single_dim_strategy(
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

purposefully not enabled yet. but I left the commented code here since I often want to switch it back on for testing

@wconstab
Copy link
Contributor Author

@pytorchbot merge -i

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Dec 20, 2025
@pytorchmergebot
Copy link
Collaborator

Merge started

Your 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

Advanced Debugging
Check the merge workflow status
here

xgz2 pushed a commit that referenced this pull request Dec 22, 2025
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>
krastogi-in pushed a commit to krastogi-in/pytorch that referenced this pull request Jan 9, 2026
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>
@github-actions github-actions bot deleted the gh/wconstab/459/head branch January 20, 2026 02:19
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/inductor ciflow/trunk Trigger trunk jobs on your pull request Merged topic: not user facing topic category

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants