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

Schedule Transferability between Intel and ARM CPU targets #5340

Closed
4 tasks
anijain2305 opened this issue Apr 15, 2020 · 3 comments
Closed
4 tasks

Schedule Transferability between Intel and ARM CPU targets #5340

anijain2305 opened this issue Apr 15, 2020 · 3 comments

Comments

@anijain2305
Copy link
Contributor

anijain2305 commented Apr 15, 2020

Relevant discuss post - https://discuss.tvm.ai/t/topi-using-x86-schedules-for-arm-conv2d/6365

Currently, TVM has different schedules for ARM and Intel for conv2d operators. The discuss post listed above shows that Intel conv2d NCHWc schedule on ARM gives better end-to-end latency compared to ARM NCHW conv2d spatial pack schedule for many TFLite networks.

However, this is just one opportunity and there are also some more ideas that we should pursue. This issue lists those potential issues and anybody interested can pick them up. This list is a result of discussions from the above post.

@FrozenGene @masahi @tqchen

@zhenhuaw-me
Copy link
Contributor

That is really good idea that we can share schedules across backends!
A few monthes ago, I had once thought that, maybe we can modularize TOPI such that some well known schedules can be shared as much as possible. For example, we can have schedules for CPU and GPU which could shared between x86/ARM/... and CUDA/OpenCL respectively. Though I am not actively working on TVM nowdays, hoping to contribute someday still.

@tqchen
Copy link
Member

tqchen commented Nov 1, 2020

@anijain2305 please check if we can close this issue either due to non-actionable status(stale) or completion

@anijain2305
Copy link
Contributor Author

I think we can close this. As Ansor comes in, we will try these different choices automatically on all platforms.

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

No branches or pull requests

3 participants