Skip to content

Commit

Permalink
[Blocking] Fix a bug on prefetch_tile creation in blocking (#990)
Browse files Browse the repository at this point in the history
  • Loading branch information
chencha3 authored Dec 18, 2024
1 parent fc16bd4 commit 65fab90
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions lib/Dialect/XeTile/Transforms/Blocking.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -702,8 +702,9 @@ class RewritePrefetchTileOp
auto convertedTiles = addPackOp(tile, convertedTileTypes,
blockSize.asArrayRef(), loc, rewriter);

for (auto [t, ty] : llvm::zip_equal(convertedTiles, convertedTileTypes)) {
rewriter.create<xetile::PrefetchTileOp>(loc, ty, t, op->getAttrs());
for (auto t : convertedTiles) {
rewriter.create<xetile::PrefetchTileOp>(loc, mlir::TypeRange(), t,
op->getAttrs());
}

rewriter.eraseOp(op);
Expand Down

0 comments on commit 65fab90

Please sign in to comment.