Skip to content
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 cpu][auto scheduler] Auto scheduler @ arm cpu can not reproduce paper's performance #7625

Closed
FrozenGene opened this issue Mar 10, 2021 · 1 comment · Fixed by #7635
Closed
Assignees

Comments

@FrozenGene
Copy link
Member

currently, in the upstream version of auto scheduler on arm cpu can not reproduce OSDI paper's performance. One issue we find is we call wrong compute for arm cpu (https://github.com/apache/tvm/blob/main/python/tvm/relay/op/strategy/arm_cpu.py#L156) and we don't enable winograd for arm cpu too. Open this issue to tracker this.

@jcf94

@jcf94 jcf94 self-assigned this Mar 10, 2021
@jcf94
Copy link
Contributor

jcf94 commented Mar 12, 2021

#7635 Add a fix on arm strategy to choose the right nwhc conv2d implementation with layout rewrite support.

Will continue to see if there's any other problem.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
2 participants