junrushao1994 opened a new pull request #9382: URL: https://github.com/apache/tvm/pull/9382
This PR enables meta schedule to extract tasks from the high-level framework, in our particular case, Relay. A base class `IntegrationContext` is introduced as the context manager for both task extraction (observer mode) and tuned task injection (injector mode): - Derived class `TaskExtraction` serves as the observer mode to extract tasks from Relay - Derived class `ApplyHistoryBest` serves as the injector mode to inject the tuned task back to Relay The primary API for task extraction is `meta_schedule.integration.extract_task`, which returns a list of tasks on both Relay and TIR-level. The PR only handles task extraction, but doesn't include further work like: - Multi-dispatching from Relay to TE/TIR - Deduplication and similarity score calculation of extracted tasks - Relay TIR operator strategy - Layout rewriting CC: @mbrookhart @mbs-octoml @sunggg @zxybazh @Hzfengsy -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
