[TVM Discuss] [Questions] How to do auto-tuning for a list of specific ops?

2020-05-28 Thread Cody H. Yu via TVM Discuss
Good questions in general. 1. This would be a problem if you have limited memory in the machine. 2. Currently AutoTVM doesn't have an official API to export tasks. At this moment, one solution I can think of is writing a script to extract necessary task information and save it to a JSON file.

[TVM Discuss] [Questions] How to do auto-tuning for a list of specific ops?

2020-05-27 Thread Sleepwalker2017 via TVM Discuss
Thank you ! I have two questions for this solution. 1. To use this API, I have to load all models at the same time, which seems to have a huge demand for memory. 2. I didn't find a way to save tasks to disk, that means if the tuning is interrupted , I have to do the whole thing from beginnin

[TVM Discuss] [Questions] How to do auto-tuning for a list of specific ops?

2020-05-27 Thread Cody H. Yu via TVM Discuss
The official solution to this question is leveraging `extract_from_multiple_programs`. You can give that API multiple models at once, and it will return a list of unique tasks. By tuning the list of tasks, your log file can be used for all models you provided. --- [Visit Topic](https://d

[TVM Discuss] [Questions] How to do auto-tuning for a list of specific ops?

2020-05-27 Thread Sleepwalker2017 via TVM Discuss
Thank you!Seems it will work to me. --- [Visit Topic](https://discuss.tvm.ai/t/how-to-do-auto-tuning-for-a-list-of-specific-ops/6817/3) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/u

[TVM Discuss] [Questions] How to do auto-tuning for a list of specific ops?

2020-05-27 Thread kalyan cimata via TVM Discuss
Hi, The implementation of function **prune_old_tasks** in this discussion https://discuss.tvm.ai/t/solved-can-we-resume-an-autotuning-session/3329 may helps you. --- [Visit Topic](https://discuss.tvm.ai/t/how-to-do-auto-tuning-for-a-list-of-specific-ops/6817/2) to respond. You are rec