Update permute removal pass to handle binary operations, and cleanup better#18256
Update permute removal pass to handle binary operations, and cleanup better#18256mcremon-meta wants to merge 4 commits intomainfrom
Conversation
Summary: As titled. Useful especially for resnet backbones, to reduce quant/dequant pressure. Reviewed By: DrJessop Differential Revision: D96683989
Summary: As titled. Should perform better and also allow removing some permutes when convolutions are also moved to channel last. Reviewed By: hsharma35 Differential Revision: D96869747
Summary: As titled. Calls into nnlib directly. Reviewed By: hsharma35 Differential Revision: D96874522
…better Summary: As titled. It is currently not cleaning up as much as it should, and the pass is only capable of handling single input cases. Result: from 9 to 1 (minimum by construction) permutes on Wake Gesture. Reviewed By: abeakkas Differential Revision: D96940254
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/18256
Note: Links to docs will display an error until the docs builds have been completed. ⏳ 4 Pending, 3 Unrelated FailuresAs of commit ad581f1 with merge base ed57040 ( FLAKY - The following job failed but was likely due to flakiness present on trunk:
BROKEN TRUNK - The following jobs 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. |
meta-cla
bot
added
the
CLA Signed
|
@mcremon-meta has exported this pull request. If you are a Meta employee, you can view the originating Diff in D96940254. |
This PR needs a
|
1 participant
Summary:
As titled. It is currently not cleaning up as much as it should, and the pass is only capable of handling single input cases.
Result: from 9 to 1 (minimum by construction) permutes on Wake Gesture.
Reviewed By: abeakkas
Differential Revision: D96940254