This is interesting work. I’m curious if the plan is that in the future auto_scheduler would not rely on any new custom Relay ops for tuning subgraphs like the example that is shown, i.e. it directly tunes a primitive function as the user designates?
I’m working on a similar thing of subgraph tuning. Now it runs with topi schedules + pattern matching + customer Relay ops, without auto_scheduler being involved yet. I’m looking to working on more complex subgraphs, and would love to know if there’s a chance to collaborate!