diff --git a/lib/Dialect/XeTile/Transforms/BlockingAnalysis.cpp b/lib/Dialect/XeTile/Transforms/BlockingAnalysis.cpp
index fad3acd18..0e0dd7fb5 100644
--- 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();