[llvm] [WIP][DAG] Add legalization handling for AVGCEIL/AVGFLOOR nodes (PR #92096)
Simon Pilgrim via llvm-commits
llvm-commits at lists.llvm.org
Thu May 16 09:56:32 PDT 2024
================
@@ -4560,8 +4560,15 @@ unsigned SelectionDAG::ComputeNumSignBits(SDValue Op, const APInt &DemandedElts,
Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
// SRA X, C -> adds C sign bits.
if (const APInt *ShAmt =
- getValidMinimumShiftAmountConstant(Op, DemandedElts))
+ getValidMinimumShiftAmountConstant(Op, DemandedElts)) {
Tmp = std::min<uint64_t>(Tmp + ShAmt->getZExtValue(), VTBits);
+ } else {
+ KnownBits KnownAmt =
+ computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
+ if (KnownAmt.isConstant() && KnownAmt.getConstant().ult(VTBits))
+ Tmp = std::min<uint64_t>(Tmp + KnownAmt.getConstant().getZExtValue(),
+ VTBits);
+ }
----------------
RKSimon wrote:
If only... its to fix a thumb2 regression as it lowers v2i64 constant as bitcast(v4i32 constant)
Once this draft has addressed all the regressions I'll turn my attention to pulling out some of these changes.
https://github.com/llvm/llvm-project/pull/92096
More information about the llvm-commits
mailing list