[Mlir-commits] [mlir] Update `LowerContractionToSMMLAPattern` to ingnore matvec (PR #88288)

Diego Caballero llvmlistbot at llvm.org
Wed Apr 10 09:20:23 PDT 2024


================
@@ -54,6 +54,8 @@ class LowerContractionToSMMLAPattern
     // Note: RHS is not transposed.
     mlir::VectorType lhsType = op.getLhsType();
     mlir::VectorType rhsType = op.getRhsType();
+    if (rhsType.getRank() < 2)
+      return failure();
----------------
dcaballe wrote:

Since 0-D vectors can also reach this point, could you add another check making sure that if any of the operands doesn't have a rank, we also bail out (+ test for that)?

https://github.com/llvm/llvm-project/pull/88288


More information about the Mlir-commits mailing list