-
Notifications
You must be signed in to change notification settings - Fork 537
Refactor TOSA backend preprocess to use visitor pattern #1099
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
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/1099
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 New FailureAs of commit e727104 with merge base a8e05cc ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
✅ Deploy Preview for resplendent-gnome-14e531 canceled.
|
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.
So clean!! :) Left some comments but lgtm.
nit - seems like there are some changes which are not purely refactor, perhaps move them to another PR?
This is great! I have no idea how you manage to read the previous code.... |
Signed-off-by: Jerry Ge <[email protected]> Change-Id: I1d7275bdc81c9ad60c27f21fa604c134aa3e3646
I don't think you can 🙂 |
|
||
# Helper function to do broadcasting | ||
# Ref: https://www.mlplatform.org/tosa/tosa_spec.html#_broadcasting | ||
def broadcastShapes(shape1, shape2): |
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.
does arm kernel supports broadcast?
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.
the lint-runner failure should not be related to this patch
|
yeah should be fixed soon, feel free to ignore |
@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
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.
I am assuming you ran e2e and also setup.sh and run.sh from examples/arm dir?
@digantdesai merged this pull request in 53a5cb6. |
No lowering logic change. Only a very large refactor patch to make the current code more modular for easier future maintenance and development.