-
Notifications
You must be signed in to change notification settings - Fork 602
Milestone2.1: Partition to_dim_order_copy op in XNN delegate #11286
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
base: main
Are you sure you want to change the base?
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/11286
Note: Links to docs will display an error until the docs builds have been completed. ❌ 2 New FailuresAs of commit 549b361 with merge base 18e4240 ( NEW FAILURES - The following jobs have failed:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
@pytorchbot label "release notes: none" |
7dc20ef
to
a8e4be4
Compare
Add a more descriptive name to the Title of this PR |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Implementation looks good to me. In addition to the test you added in in the pass tests, can you create a test file for to_copy under backends/xnnpack/test/ops? It might be good to add cover the following cases:
- to(channels_last) before linear.
- to(contiguous) before conv.
- Convert both dtype and dim order (for example, .to(torch.float, memory_format=torch.channels_last) with non-float inputs).
74a7331
to
d195f85
Compare
@leafs1 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
d029d18
to
3e1a4e3
Compare
Summary
Add to_dim_order_copy op to the partitioner in XNN delegate and delegate dim order conversions (to_dim_order_copy op) manually initiated by user using .to(memory_format=)
Test plan
Confirmed expected output when having user specified dim order conversions as well as appropriate partitioning