[llvm] [AArch64] Lower scalable i1 vector add reduction to cntp (PR #99031)

Sander de Smalen via llvm-commits llvm-commits at lists.llvm.org
Thu Jul 18 09:20:09 PDT 2024


================
@@ -27469,6 +27469,21 @@ SDValue AArch64TargetLowering::LowerReductionToSVE(unsigned Opcode,
     VecOp = convertToScalableVector(DAG, ContainerVT, VecOp);
   }
 
+  // Lower VECREDUCE_ADD of nxv2i1-nxv16i1 to CNTP rather than UADDV.
+  if (ScalarOp.getOpcode() == ISD::VECREDUCE_ADD &&
+      VecOp.getOpcode() == ISD::ZERO_EXTEND) {
+    SDValue Vec = VecOp.getOperand(0);
+    EVT VecVT = Vec.getValueType();
----------------
sdesmalen-arm wrote:

nit: VecVT has only one use and can be inlined.

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


More information about the llvm-commits mailing list