Add proper vnni check

libxsmm · Jan 9, 2025 · 6513c14 · 6513c14
1 parent aac340a
commit 6513c14
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/TPP/Conversion/ConvertLinalgToXsmm/ConvertLinalgToXsmm.cpp b/lib/TPP/Conversion/ConvertLinalgToXsmm/ConvertLinalgToXsmm.cpp
@@ -1113,7 +1113,9 @@ struct ConvertGenericToVnniMatmulLikeOp
     // by the earlier check) that splits the K dim in the shape.
     std::optional<int64_t> vnniFactor =
         vnni::utils::getVnniBlockingFactor(bufferB.getType());
-    assert(vnniFactor && "Must be in VNNI format");
+    if (!vnniFactor)
+      return rewriter.notifyMatchFailure(genericOp,
+                                         "failed to determine VNNI factor");
     int64_t k =
         cast<ShapedType>(bufferA.getType()).getShape()[kPos] * *vnniFactor;
     int64_t batch = 0;