mbs-octoml opened a new pull request, #12078: URL: https://github.com/apache/tvm/pull/12078
See https://github.com/apache/tvm-rfcs/blob/main/rfcs/0062-collage.md. We complete the PartitionRule sub-class hierarchy with the addition of CombinePartitionRule, which allows disjoint candidate partitions to be unioned based on simple rules. - By TOpPattern kind, eg a kOutElemwiseFusable and kBroadcast. - A tuple argument with injective fields. - The projection from an injective group (obviously of tuple type) - Combinations of the above. These let us mimic many common fusion strategies, including TVMs, so that the candidates explored during Collage search are as large as possible to expose possible fusion opportunities but no larger. Also completes CandidatePartition with the EstimateCost method, which is used during search to construct a stand-alone IRModule for latency estimation. Finish units tests for PartitionRule and CandidatePartition. -- 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]
