[llvm] [AArch64][NEON] Lower fixed-width add partial reductions to dot product (PR #107078)

Sam Tebbs via llvm-commits llvm-commits at lists.llvm.org
Thu Sep 5 06:31:55 PDT 2024


================
@@ -21781,7 +21782,12 @@ SDValue tryLowerPartialReductionToDot(SDNode *N,
              Intrinsic::experimental_vector_partial_reduce_add &&
          "Expected a partial reduction node");
 
-  if (!Subtarget->isSVEorStreamingSVEAvailable())
+  if (!Subtarget->isSVEorStreamingSVEAvailable() &&
+      !Subtarget->isNeonAvailable())
+    return SDValue();
+
+  // Fixed-width requires the dotprod feature, both for Neon and SVE
+  if (!N->getValueType(0).isScalableVT() && !Subtarget->hasDotProd())
----------------
SamTebbs33 wrote:

Done.

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


More information about the llvm-commits mailing list