[mlir][Vector] Support 0-D vectors in FMAOp

Reviewed By: ftynse Differential Revision: https://reviews.llvm.org/D115742
2022-08-24 07:59:59 -07:00
parent 540a13652f
commit ab45a4329b
5 changed files with 30 additions and 11 deletions
--- a/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp
+++ b/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp
@@ -746,7 +746,7 @@ public:
  matchAndRewrite(vector::FMAOp fmaOp, OpAdaptor adaptor,
                  ConversionPatternRewriter &rewriter) const override {
    VectorType vType = fmaOp.getVectorType();
-    if (vType.getRank() != 1)
+    if (vType.getRank() > 1)
      return failure();
    rewriter.replaceOpWithNewOp<LLVM::FMulAddOp>(
        fmaOp, adaptor.getLhs(), adaptor.getRhs(), adaptor.getAcc());