comaniac opened a new pull request #8695: URL: https://github.com/apache/tvm/pull/8695
The AutoScheduler FLOP estimator now mis-estimates the following case to be 0 FLOPS: ``` tir.if_then_else(A[i] > 0, A[i], 0) ``` This is because the expression `A[i] > 0` has output dtype `uint` while the input dtypes are `float32`. In this case the current FLOP estimater ignores the comparison flops. The original concern behinds this logic is to avoid counting index calculation, but we need to make it more general to cover the above case. cc @merrymercy -- 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]
