Reland "Arm backend: Add real implementation for TOSA dialect ops (re-land)" v2#20674
Reland "Arm backend: Add real implementation for TOSA dialect ops (re-land)" v2#20674rascani wants to merge 1 commit into
Conversation
… ops (re…" This reverts commit 5909e23.
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/20674
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit d87e5e5 with merge base 5909e23 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
|
@rascani has imported this pull request. If you are a Meta employee, you can view this in D110362915. |
Reverts #20670 and relands #20537
In a first commit,
Arm backend: Add infra for real implementations of Tosa ops
Tested by applying the infra to the avg_pool2d Tosa dialect op.
The rewrite_avg_pool2d tests can now be ran to verify that
the produced Tosa is correct. To make it completely correct,
two additional passes need to be added to the test.
Then,
Arm backend: Add real impls to all TOSA dialect ops
Additionally,
Remove special case in ComputeOpsAOT pass for such ops,
since they can now be executed.
Start running the model in tests were this was previously
impossible due to ops not having a real impl.
cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell