Closed Anndrey24 closed 1 month ago
cc @Lunderberg
We decided to roll back to the initial version of this patch since a large number of Relax tests fail when division by zero is disabled in the rewrite rules (so dealing with this is out of scope for this patch).
Thanks @Anndrey24!
This commit introduces rewrite rules for indices which can arise from splitting axes by scalable factors (e.g.
xo, xi = sch.split(x, factors = [None, 8 * T.vscale()])
):The rewrites help prove checks needed by
sch.tensorize()
(e.g. CompareBufferRegion).cc @ekalda @lhutton1