llvm/llvm-project

Wrong types checked at DAGCombiner::visitSRA(SDNode *N)

Open

#153,543 创建于 2025年8月14日

在 GitHub 查看
 (7 评论) (0 反应) (1 负责人)C++ (10,782 fork)batch import
good first issuellvm:SelectionDAG

仓库指标

Star
 (26,378 star)
PR 合并指标
 (平均合并 1天 2小时) (30 天内合并 1,000 个 PR)

描述

The following check is performed at DAGCombiner::visitSRA(SDNode *N)

if ((ShiftAmt > 0) &&
          TLI.isOperationLegalOrCustom(ISD::SIGN_EXTEND, **TruncVT**) &&
          TLI.isOperationLegalOrCustom(ISD::TRUNCATE, **VT**) &&
          TLI.isTruncateFree(VT, TruncVT)) {
        SDValue Amt = DAG.getShiftAmountConstant(ShiftAmt, VT, DL);
        SDValue Shift = DAG.getNode(ISD::SRL, DL, VT,
                                    N0.getOperand(0), Amt);
        SDValue Trunc = DAG.getNode(ISD::TRUNCATE, DL, TruncVT,
                                    Shift);
        return DAG.getNode(ISD::SIGN_EXTEND, DL,
                           N->getValueType(0), Trunc);
      }

but it should be

if ((ShiftAmt > 0) &&
          TLI.isOperationLegalOrCustom(ISD::SIGN_EXTEND, **VT**) &&
          TLI.isOperationLegalOrCustom(ISD::TRUNCATE, **TruncVT**) &&
          TLI.isTruncateFree(VT, TruncVT)) {
        SDValue Amt = DAG.getShiftAmountConstant(ShiftAmt, VT, DL);
        SDValue Shift = DAG.getNode(ISD::SRL, DL, VT,
                                    N0.getOperand(0), Amt);
        SDValue Trunc = DAG.getNode(ISD::TRUNCATE, DL, TruncVT,
                                    Shift);
        return DAG.getNode(ISD::SIGN_EXTEND, DL,
                           N->getValueType(0), Trunc);
      }

this mistake can cause operation marked with expand to reach selection

贡献者指南