[Blocking] Mark vector.store and vector.load as legal op (#911)

Mark vector.store and vector.load as legal op
intel · Oct 3, 2024 · f31a332 · f31a332
1 parent 44a09ea
commit f31a332
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/Dialect/XeTile/Transforms/BlockingRewrite.cpp b/lib/Dialect/XeTile/Transforms/BlockingRewrite.cpp
@@ -886,6 +886,8 @@ class NewXeTileBlockingPass
     target.addLegalOp<xetile::TilePackOp>();
     target.addLegalOp<xetile::TileUnpackOp>();
     target.addLegalOp<mlir::vector::ShapeCastOp>();
+    target.addLegalOp<mlir::vector::StoreOp>();
+    target.addLegalOp<mlir::vector::LoadOp>();
 
     target.addDynamicallyLegalOp<mlir::arith::ConstantOp>(
         [&](mlir::arith::ConstantOp op) -> bool {