limits to 2D vector ops only (#941)

intel · Oct 24, 2024 · c496ee1 · c496ee1
1 parent 52cadab
commit c496ee1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/Dialect/XeTile/Transforms/BlockingAnalysis.cpp b/lib/Dialect/XeTile/Transforms/BlockingAnalysis.cpp
@@ -563,7 +563,7 @@ void BlockingAnalysisImpl::visitVectorizableOp(
     return;
 
   auto type = mlir::dyn_cast<mlir::VectorType>(op->getResult(0).getType());
-  if (!type)
+  if (!type || type.getRank() != 2)
     return;
 
   auto lattice = results[0]->getValue();