llvm/llvm-project

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

開放

#153,543 建立於 2025年8月14日

 (7 則留言) (0 個反應) (1 位負責人)C++ (10,782 個分叉)batch import
good first issuellvm:SelectionDAG

倉庫指標

星標
 (26,378 顆星)
PR 合併指標
 (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

貢獻者指南