[llvm] [SelectionDAG] Fold constant min/max vector reductions (PR #209190)
Luke Lau via llvm-commits
llvm-commits at lists.llvm.org
Mon Jul 13 23:26:37 PDT 2026
================
@@ -7802,19 +7806,26 @@ SDValue SelectionDAG::FoldConstantArithmetic(unsigned Opcode, const SDLoc &DL,
if (Opcode == ISD::BITCAST)
return SDValue();
- // Constant fold VECREDUCE_ADD with a BUILD_VECTOR of integer constants.
- if (Opcode == ISD::VECREDUCE_ADD && ISD::isBuildVectorOfConstantSDNodes(N1.getNode())) {
+ // Constant fold integer vector reductions with constant BUILD_VECTORs.
+ if ((Opcode == ISD::VECREDUCE_ADD || Opcode == ISD::VECREDUCE_SMAX ||
+ Opcode == ISD::VECREDUCE_SMIN || Opcode == ISD::VECREDUCE_UMAX ||
+ Opcode == ISD::VECREDUCE_UMIN) &&
+ ISD::isBuildVectorOfConstantSDNodes(N1.getNode())) {
unsigned EltBits = N1.getValueType().getScalarSizeInBits();
- APInt Acc = APInt::getZero(EltBits);
+ std::optional<APInt> Acc;
----------------
lukel97 wrote:
I would rewrite this to start with the identity element for BaseOpcode, so you don't need the `std::optional`. I think you can use `SelectionDAG::getIdentityElement`
https://github.com/llvm/llvm-project/pull/209190
More information about the llvm-commits
mailing list