zxybazh opened a new pull request #9761: URL: https://github.com/apache/tvm/pull/9761
This PR is part of the stage M3c of the meta schedule project (#8473). The architecture is re-designed by Junru and Xiyou. Here we introduce the `ScheduleRule` class to process tir blocks while generating new design spaces via `PostOrderApply` space generator. The post-order-apply space generator collects all the blocks in IRModule and apply the rules in post order. Here we also allow custmization of schedule rules using a python side class `PyScheduleRule`. Note that we introduced method definition check `check_override` to make sure certain methods are overrided so that an error will be thown upon definition of an incomplete PyClass. Unittests are also introduced in this PR. Thanks to all co-authors for contributing! Co-authored-by: Junru Shao <<[email protected]>> Co-authored-by: Bohan Hou <<[email protected]>> Co-authored-by: Ruihang Lai <<[email protected]>> Co-authored-by: Hongyi Jin <<[email protected]>> Co-authored-by: Wuwei Lin <<[email protected]>> Co-authored-by: Siyuan Feng <<[email protected]>> -- 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]
