farshidsp opened a new pull request, #14379: URL: https://github.com/apache/tvm/pull/14379
FlopEstimator is run as part of the Meta Schedule as an estimate for which primfunc to spend time on; e.g. which primfunc to send more trials on. In a case that the for loop extent is an expression, it’s getting stuck on the calculation of the FLOPS. This PR add a function annotation to the main primfunc and then have FlopEstimator check for this annotation before visiting the function. An if the annotation exists, just use the handcoded flops instead of trying to estimate. @Lunderberg @csullivan -- 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]
