Skip to content

[XNNPACK][Partitioner] enable src based partitioner #4795

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Aug 20, 2024

Conversation

kirklandsign
Copy link
Contributor

Stack from ghstack (oldest at bottom):

In order to maintain parity with the current to_edge and to_backend lowering flow, we need to support source based partitioning.

We apply source-based partitioning to the AddMMConfig to partition all the nodes surrounding addmm so that it can be recomposed internally. While this is fine for to_edge and to_backend flow. For more robust flow, we will not have to use this when running to_edge_transform_and_lower.

Differential Revision: D61250576

Pull Request resolved: #4762

mcr229 added 3 commits August 16, 2024 15:47
In order to maintain parity with the current to_edge and to_backend lowering flow, we need to support source based partitioning.

We apply source-based partitioning to the AddMMConfig to partition all the nodes surrounding addmm so that it can be recomposed internally. While this is fine for to_edge and to_backend flow. For more robust flow, we will not have to use this when running to_edge_transform_and_lower.

Differential Revision: [D61250576](https://our.internmc.facebook.com/intern/diff/D61250576/)

[ghstack-poisoned]
…rtitioner"

In order to maintain parity with the current to_edge and to_backend lowering flow, we need to support source based partitioning.

We apply source-based partitioning to the AddMMConfig to partition all the nodes surrounding addmm so that it can be recomposed internally. While this is fine for to_edge and to_backend flow. For more robust flow, we will not have to use this when running to_edge_transform_and_lower.

Differential Revision: [D61250576](https://our.internmc.facebook.com/intern/diff/D61250576/)

[ghstack-poisoned]
In order to maintain parity with the current to_edge and to_backend lowering flow, we need to support source based partitioning.

We apply source-based partitioning to the AddMMConfig to partition all the nodes surrounding addmm so that it can be recomposed internally. While this is fine for to_edge and to_backend flow. For more robust flow, we will not have to use this when running to_edge_transform_and_lower.

Differential Revision: [D61250576](https://our.internmc.facebook.com/intern/diff/D61250576/)

[ghstack-poisoned]
Copy link

pytorch-bot bot commented Aug 20, 2024

🔗 Helpful Links

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

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

✅ No Failures

As of commit 4139a40 with merge base 9e478e8 (image):
💚 Looks good so far! There are no failures yet. 💚

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

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Aug 20, 2024
@kirklandsign kirklandsign merged commit 4a27a53 into main Aug 20, 2024
64 of 67 checks passed
@kirklandsign kirklandsign deleted the gh/mcr229/1/head branch August 20, 2024 18:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants