[AArch64] Fix Fold of Compare with Right-shifted Value

This commit folds (setcc ne (lshr x c) 0) for 64-bit types and constants
>= 32. This fold already existed for other types or smaller constants
but was not applicable to 64-bit types and constants >= 32 due to a
comparison of the constant c with the bit size of the setcc operation.
The type of this operation is legalized to i32, which does not
necessarily match the type of the lshr operation. Use the bit size of
the type of the lshr operation instead for the comparison.

Fixes #122380.

Author: mskamp

llvm/lib/Target/AArch64/AArch64ISelLowering.cpp

@@ -25070,10 +25070,10 @@ static SDValue performSETCCCombine(SDNode *N,
   // setcc (srl x, imm), 0, ne ==> setcc (and x, (-1 << imm)), 0, ne
   if (Cond == ISD::SETNE && isNullConstant(RHS) &&
       LHS->getOpcode() == ISD::SRL && isa<ConstantSDNode>(LHS->getOperand(1)) &&
-      LHS->getConstantOperandVal(1) < VT.getScalarSizeInBits() &&
       LHS->hasOneUse()) {
     EVT TstVT = LHS->getValueType(0);
-    if (TstVT.isScalarInteger() && TstVT.getFixedSizeInBits() <= 64) {
+    if (TstVT.isScalarInteger() && TstVT.getFixedSizeInBits() <= 64 &&
+        LHS->getConstantOperandVal(1) < TstVT.getFixedSizeInBits()) {
       // this pattern will get better opt in emitComparison
       uint64_t TstImm = -1ULL << LHS->getConstantOperandVal(1);
       SDValue TST = DAG.getNode(ISD::AND, DL, TstVT, LHS->getOperand(0),

llvm/test/CodeGen/AArch64/shift-const-ne-0.ll

@@ -69,8 +69,7 @@ define i1 @lsr_31_ne_0_64(i64 %x) {
 define i1 @lsr_32_ne_0_64(i64 %x) {
 ; CHECK-LABEL: lsr_32_ne_0_64:
 ; CHECK:       // %bb.0:
-; CHECK-NEXT:    lsr x8, x0, #32
-; CHECK-NEXT:    cmp x8, #0
+; CHECK-NEXT:    tst x0, #0xffffffff00000000
 ; CHECK-NEXT:    cset w0, ne
 ; CHECK-NEXT:    ret
   %shr = lshr i64 %x, 32
@@ -81,8 +80,7 @@ define i1 @lsr_32_ne_0_64(i64 %x) {
 define i1 @lsr_33_ne_0_64(i64 %x) {
 ; CHECK-LABEL: lsr_33_ne_0_64:
 ; CHECK:       // %bb.0:
-; CHECK-NEXT:    lsr x8, x0, #33
-; CHECK-NEXT:    cmp x8, #0
+; CHECK-NEXT:    tst x0, #0xfffffffe00000000
 ; CHECK-NEXT:    cset w0, ne
 ; CHECK-NEXT:    ret
   %shr = lshr i64 %x, 33
@@ -93,8 +91,7 @@ define i1 @lsr_33_ne_0_64(i64 %x) {
 define i1 @lsr_62_ne_0_64(i64 %x) {
 ; CHECK-LABEL: lsr_62_ne_0_64:
 ; CHECK:       // %bb.0:
-; CHECK-NEXT:    lsr x8, x0, #62
-; CHECK-NEXT:    cmp x8, #0
+; CHECK-NEXT:    tst x0, #0xc000000000000000
 ; CHECK-NEXT:    cset w0, ne
 ; CHECK-NEXT:    ret
   %shr = lshr i64 %x, 62