Enforce vnni shape to be mod 2

libxsmm · Jan 16, 2025 · c7d5a5d · c7d5a5d
1 parent 9aa7b9b
commit c7d5a5d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/TPP/Transforms/Utils/VNNIUtils.cpp b/lib/TPP/Transforms/Utils/VNNIUtils.cpp
@@ -125,6 +125,9 @@ bool isInVnniLayout(int64_t expectedRank, ShapedType shape,
   if (shape.getRank() != expectedRank || !shape.getElementType().isBF16())
     return false;
 
+  if (shape.getShape().back() % 2 != 0)
+    return false;
+
   if (blockingFactor && shape.getShape().back() != *blockingFactor)
     return false;