MasterJH5574 opened a new pull request #9814: URL: https://github.com/apache/tvm/pull/9814
This PR changes a behavior in schedule primitive decompose-reduction. Prior to this PR, loops outside the newly created init block always have the for-kind "serial". And this PR changes this behavior, letting the new loops inherit the for-kinds from the old loops. Several reasons for this change: * If the new loops don't inherit the for-kinds, their for-kinds is impossible to be changed once after the decompose-reduction. This is because after decompose-reduction, the init block and the update block write to a same buffer. Thus the compact-dataflow property is no longer satisfied (more specifically, the "dominant" property is no longer satisfied). And in this case, we cannot apply vectorization or parallelization to those loops. * From the perspective of performance, it makes sense to let the new loops have the same for-kinds as the original loops. * TE has this behavior. --- cc @spectrometerHBH @junrushao1994 @Hzfengsy -- 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]
