MasterJH5574 opened a new pull request, #14595: URL: https://github.com/apache/tvm/pull/14595
PR #14182 introduces the default GPU schedule pass, which makes uses of the BlockCollector in Meta-Schedule. When requested, the BlockCollector goes through every function in the input IRModule, while in the case of applying default GPU schedule where we apply schedule to one PrimFunc at a time, going through every PrimFunc is wasteful. Therefore, this PR enhances the BlockCollector so that it only focuses on the function being worked on. -- 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]
