-
Notifications
You must be signed in to change notification settings - Fork 682
Arm backend: Backend test serializes and uses EthosUQuant on Ethos-U flows #14817
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
Arm backend: Backend test serializes and uses EthosUQuant on Ethos-U flows #14817
Conversation
…s-U flows This make Ethos-U flows work. Signed-off-by: Zingo Andersen <[email protected]> Change-Id: Ib665857761f158b4a5b966d135ba335f28846109
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/14817
Note: Links to docs will display an error until the docs builds have been completed. ❌ 2 New FailuresAs of commit aff7b55 with merge base 3557edf ( NEW FAILURES - The following jobs have failed:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
kwargs["compile_spec"] = compile_spec | ||
return ArmTester(*args, **kwargs) | ||
|
||
support_serialize = not isinstance(compile_spec, TosaCompileSpec) |
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.
we should be able to run on tosa reference model, no?
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.
We do not have a "runner" for it. We only have a special testflow.
Enabling serilize right now causes more fails. Need to investigate more so I left it as a future exercise
) | ||
|
||
ARM_ETHOS_U55_FLOW = _create_tosa_flow( | ||
ARM_ETHOS_U55_FLOW = _create_arm_flow( |
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.
When running the model won't this flow run into long runtime issues?
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.
runs kind of OK on my laptop, modeltest took about 30-40min I think. Lets see :)
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.
Stamping to unblock, left some comments. Thanks @zingo !
Summary
Serialize and quantize automaticaly when possible. This make Ethos-U flows work.
Test plan
This is runned by the backed suit testing for Ethos-U
cc @freddan80 @per @oscarandersson8218 @digantdesai