[mlir][linalg] Fix tensor tiling together with interchange
In `linalg::tileConsumerAndFuseProducers`, there are two levels of tiling and fusion; we partition the tile sizes and only use one half for each of them. The partition is using the first non-parallel dimension *after* interchange as the boundary. However, concrete tiling happens *together with* loop interchange, so we still need to provide the partial tile sizes *before* the interchange. Otherwise, there will be inconsistency, which is what this patch is to fix. Reviewed By: ThomasRaoux Differential Revision: https://reviews.llvm.org/D129804
Loading
Please sign in to comment