llvm/llvm/test/CodeGen/X86/apx/shift-eflags.ll

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc < %s -mtriple=x86_64-- -mattr=+ndd | FileCheck %s

; Use shift eflags result when it won't cause stalls

; ashr by constant - use sarl eflags result
define i32 @ashr_const(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: ashr_const:
; CHECK:       # %bb.0:
; CHECK-NEXT:    sarl $14, %edi
; CHECK-NEXT:    cmovel %edx, %ecx, %eax
; CHECK-NEXT:    retq
  %s = ashr i32 %a0, 14
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; lshr by constant - simplify to test
define i32 @lshr_const(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: lshr_const:
; CHECK:       # %bb.0:
; CHECK-NEXT:    testl $-16384, %edi # imm = 0xC000
; CHECK-NEXT:    cmovel %edx, %ecx, %eax
; CHECK-NEXT:    retq
  %s = lshr i32 %a0, 14
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; shl by constant - simplify to test
define i32 @shl_const(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: shl_const:
; CHECK:       # %bb.0:
; CHECK-NEXT:    testl $262143, %edi # imm = 0x3FFFF
; CHECK-NEXT:    cmovel %edx, %ecx, %eax
; CHECK-NEXT:    retq
  %s = shl i32 %a0, 14
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; ashr by constant and using shift result - use sarl eflags result
define i32 @ashr_const_self_select(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: ashr_const_self_select:
; CHECK:       # %bb.0:
; CHECK-NEXT:    sarl $14, %edi, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %s = ashr i32 %a0, 14
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; lshr by constant and using shift result - use shrl eflags result
define i32 @lshr_const_self_select(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: lshr_const_self_select:
; CHECK:       # %bb.0:
; CHECK-NEXT:    shrl $14, %edi, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %s = lshr i32 %a0, 14
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; lshr by constant and using result - use shll eflags result
define i32 @shl_const_self_select(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: shl_const_self_select:
; CHECK:       # %bb.0:
; CHECK-NEXT:    shll $14, %edi, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %s = shl i32 %a0, 14
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; ashr by 1 - use sarl eflags result
define i32 @ashr_const1(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: ashr_const1:
; CHECK:       # %bb.0:
; CHECK-NEXT:    sarl %edi
; CHECK-NEXT:    cmovel %edx, %ecx, %eax
; CHECK-NEXT:    retq
  %s = ashr i32 %a0, 1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; lshr by 1 - simplify to test
define i32 @lshr_const1(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: lshr_const1:
; CHECK:       # %bb.0:
; CHECK-NEXT:    testl $-2, %edi
; CHECK-NEXT:    cmovel %edx, %ecx, %eax
; CHECK-NEXT:    retq
  %s = lshr i32 %a0, 1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; shl by 1 - simplify to test
define i32 @shl_const1(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: shl_const1:
; CHECK:       # %bb.0:
; CHECK-NEXT:    testl $2147483647, %edi # imm = 0x7FFFFFFF
; CHECK-NEXT:    cmovel %edx, %ecx, %eax
; CHECK-NEXT:    retq
  %s = shl i32 %a0, 1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; ashr by 1 and using shift result - use sarl eflags result
define i32 @ashr_const1_self_select(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: ashr_const1_self_select:
; CHECK:       # %bb.0:
; CHECK-NEXT:    sarl %edi, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %s = ashr i32 %a0, 1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; lshr by 1 and using shift result - use shrl eflags result
define i32 @lshr_const1_self_select(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: lshr_const1_self_select:
; CHECK:       # %bb.0:
; CHECK-NEXT:    shrl %edi, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %s = lshr i32 %a0, 1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; lshr by 1 and using result - use addl eflags result
define i32 @shl_const1_self_select(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: shl_const1_self_select:
; CHECK:       # %bb.0:
; CHECK-NEXT:    addl %edi, %edi, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %s = shl i32 %a0, 1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; ashr by variable - use separate test
define i32 @ashr_var(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: ashr_var:
; CHECK:       # %bb.0:
; CHECK-NEXT:    movl %ecx, %eax
; CHECK-NEXT:    movl %esi, %ecx
; CHECK-NEXT:    # kill: def $cl killed $cl killed $ecx
; CHECK-NEXT:    sarl %cl, %edi
; CHECK-NEXT:    testl %edi, %edi
; CHECK-NEXT:    cmovel %edx, %eax
; CHECK-NEXT:    retq
  %s = ashr i32 %a0, %a1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; lshr by variable - use separate test
define i32 @lshr_var(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: lshr_var:
; CHECK:       # %bb.0:
; CHECK-NEXT:    movl %ecx, %eax
; CHECK-NEXT:    movl %esi, %ecx
; CHECK-NEXT:    # kill: def $cl killed $cl killed $ecx
; CHECK-NEXT:    shrl %cl, %edi
; CHECK-NEXT:    testl %edi, %edi
; CHECK-NEXT:    cmovel %edx, %eax
; CHECK-NEXT:    retq
  %s = lshr i32 %a0, %a1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; shl by variable - use separate test
define i32 @shl_var(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: shl_var:
; CHECK:       # %bb.0:
; CHECK-NEXT:    movl %ecx, %eax
; CHECK-NEXT:    movl %esi, %ecx
; CHECK-NEXT:    # kill: def $cl killed $cl killed $ecx
; CHECK-NEXT:    shll %cl, %edi
; CHECK-NEXT:    testl %edi, %edi
; CHECK-NEXT:    cmovel %edx, %eax
; CHECK-NEXT:    retq
  %s = shl i32 %a0, %a1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; ashr by variable and using result - use separate test
define i32 @ashr_var_self_select(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: ashr_var_self_select:
; CHECK:       # %bb.0:
; CHECK-NEXT:    movl %esi, %ecx
; CHECK-NEXT:    # kill: def $cl killed $cl killed $ecx
; CHECK-NEXT:    sarl %cl, %edi, %eax
; CHECK-NEXT:    testl %eax, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %s = ashr i32 %a0, %a1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; lshr by variable and using result - use separate test
define i32 @lshr_var_self_select(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: lshr_var_self_select:
; CHECK:       # %bb.0:
; CHECK-NEXT:    movl %esi, %ecx
; CHECK-NEXT:    # kill: def $cl killed $cl killed $ecx
; CHECK-NEXT:    shrl %cl, %edi, %eax
; CHECK-NEXT:    testl %eax, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %s = lshr i32 %a0, %a1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; shl by variable and using result - use separate test
define i32 @shl_var_self_select(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: shl_var_self_select:
; CHECK:       # %bb.0:
; CHECK-NEXT:    movl %esi, %ecx
; CHECK-NEXT:    # kill: def $cl killed $cl killed $ecx
; CHECK-NEXT:    shll %cl, %edi, %eax
; CHECK-NEXT:    testl %eax, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %s = shl i32 %a0, %a1
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; ashr by non-zero variable - use separate test
define i32 @ashr_var_amt_never_zero(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: ashr_var_amt_never_zero:
; CHECK:       # %bb.0:
; CHECK-NEXT:    movl %ecx, %eax
; CHECK-NEXT:    orb $1, %sil, %cl
; CHECK-NEXT:    sarl %cl, %edi
; CHECK-NEXT:    testl %edi, %edi
; CHECK-NEXT:    cmovel %edx, %eax
; CHECK-NEXT:    retq
  %a = or i32 %a1, 1
  %s = ashr i32 %a0, %a
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; lshr by non-zero variable - use separate test
define i32 @lshr_var_amt_never_zero(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: lshr_var_amt_never_zero:
; CHECK:       # %bb.0:
; CHECK-NEXT:    movl %ecx, %eax
; CHECK-NEXT:    orb $1, %sil, %cl
; CHECK-NEXT:    shrl %cl, %edi
; CHECK-NEXT:    testl %edi, %edi
; CHECK-NEXT:    cmovel %edx, %eax
; CHECK-NEXT:    retq
  %a = or i32 %a1, 1
  %s = lshr i32 %a0, %a
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; shl by non-zero variable - use separate test
define i32 @shl_var_amt_never_zero(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: shl_var_amt_never_zero:
; CHECK:       # %bb.0:
; CHECK-NEXT:    movl %ecx, %eax
; CHECK-NEXT:    orb $1, %sil, %cl
; CHECK-NEXT:    shll %cl, %edi
; CHECK-NEXT:    testl %edi, %edi
; CHECK-NEXT:    cmovel %edx, %eax
; CHECK-NEXT:    retq
  %a = or i32 %a1, 1
  %s = shl i32 %a0, %a
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %a2, i32 %a3
  ret i32 %r
}

; ashr by non-zero variable and using result - use separate test
define i32 @ashr_var_self_select_amt_never_zero(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: ashr_var_self_select_amt_never_zero:
; CHECK:       # %bb.0:
; CHECK-NEXT:    orb $1, %sil, %cl
; CHECK-NEXT:    shrl %cl, %edi, %eax
; CHECK-NEXT:    testl %eax, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %a = or i32 %a1, 1
  %s = lshr i32 %a0, %a
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; lshr by non-zero variable and using result - use separate test
define i32 @lshr_var_self_select_amt_never_zero(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: lshr_var_self_select_amt_never_zero:
; CHECK:       # %bb.0:
; CHECK-NEXT:    orb $1, %sil, %cl
; CHECK-NEXT:    shrl %cl, %edi, %eax
; CHECK-NEXT:    testl %eax, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %a = or i32 %a1, 1
  %s = lshr i32 %a0, %a
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}

; shl by non-zero variable and using result - use separate test
define i32 @shl_var_self_select_amt_never_zero(i32 %a0, i32 %a1, i32 %a2, i32 %a3) {
; CHECK-LABEL: shl_var_self_select_amt_never_zero:
; CHECK:       # %bb.0:
; CHECK-NEXT:    orb $1, %sil, %cl
; CHECK-NEXT:    shrl %cl, %edi, %eax
; CHECK-NEXT:    testl %eax, %eax
; CHECK-NEXT:    cmovnel %edx, %eax
; CHECK-NEXT:    retq
  %a = or i32 %a1, 1
  %s = lshr i32 %a0, %a
  %c = icmp eq i32 %s, 0
  %r = select i1 %c, i32 %s, i32 %a2
  ret i32 %r
}