[llvm] [AArch64][SVE] Improve code quality of vector unsigned/signed add reductions. (PR #97339)
Sander de Smalen via llvm-commits
llvm-commits at lists.llvm.org
Mon Jul 15 06:20:03 PDT 2024
================
@@ -17455,6 +17456,77 @@ static SDValue performVecReduceAddCombineWithUADDLP(SDNode *N,
return DAG.getNode(ISD::VECREDUCE_ADD, DL, MVT::i32, UADDLP);
}
+// Turn [sign|zero]_extend(vecreduce_add()) into SVE's SADDV|UADDV
+// instructions.
+static SDValue
+performVecReduceAddExtCombine(SDNode *N, TargetLowering::DAGCombinerInfo &DCI,
+ const AArch64TargetLowering &TLI) {
+ if (N->getOperand(0).getOpcode() != ISD::ZERO_EXTEND &&
+ N->getOperand(0).getOpcode() != ISD::SIGN_EXTEND)
+ return SDValue();
+ bool IsSigned = N->getOperand(0).getOpcode() == ISD::SIGN_EXTEND;
+
+ SelectionDAG &DAG = DCI.DAG;
+ auto &Subtarget = DAG.getSubtarget<AArch64Subtarget>();
+ SDValue VecOp = N->getOperand(0).getOperand(0);
+ SDLoc DL(N);
+
+ bool IsScalableType = VecOp.getValueType().isScalableVector();
----------------
sdesmalen-arm wrote:
All places where you use `IsScalableType` you invert its value first. So perhaps it makes more sense to have a variable named `IsFixedLengthType = VecOp.getValueType().isFixedLengthVector();`.
https://github.com/llvm/llvm-project/pull/97339
More information about the llvm-commits
mailing list