llvm/llvm/test/CodeGen/AArch64/logical_shifted_reg.ll

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc -verify-machineinstrs < %s -mtriple=aarch64-none-linux-gnu | FileCheck %s

@var1_32 = global i32 0
@var2_32 = global i32 0

@var1_64 = global i64 0
@var2_64 = global i64 0

define void @logical_32bit() minsize {
; CHECK-LABEL: logical_32bit:
; CHECK:       // %bb.0:
; CHECK-NEXT:    adrp x8, :got:var1_32
; CHECK-NEXT:    adrp x9, :got:var2_32
; CHECK-NEXT:    ldr x8, [x8, :got_lo12:var1_32]
; CHECK-NEXT:    ldr x9, [x9, :got_lo12:var2_32]
; CHECK-NEXT:    ldr w10, [x8]
; CHECK-NEXT:    ldr w9, [x9]
; CHECK-NEXT:    and w11, w10, w9
; CHECK-NEXT:    bic w12, w10, w9
; CHECK-NEXT:    str w11, [x8]
; CHECK-NEXT:    orr w11, w10, w9
; CHECK-NEXT:    str w12, [x8]
; CHECK-NEXT:    orn w12, w10, w9
; CHECK-NEXT:    str w11, [x8]
; CHECK-NEXT:    eor w11, w10, w9
; CHECK-NEXT:    str w12, [x8]
; CHECK-NEXT:    eon w12, w9, w10
; CHECK-NEXT:    str w11, [x8]
; CHECK-NEXT:    and w11, w10, w9, lsl #31
; CHECK-NEXT:    str w12, [x8]
; CHECK-NEXT:    bic w12, w10, w9, lsl #31
; CHECK-NEXT:    str w11, [x8]
; CHECK-NEXT:    orr w11, w10, w9, lsl #31
; CHECK-NEXT:    str w12, [x8]
; CHECK-NEXT:    orn w12, w10, w9, lsl #31
; CHECK-NEXT:    str w11, [x8]
; CHECK-NEXT:    eor w11, w10, w9, lsl #31
; CHECK-NEXT:    str w12, [x8]
; CHECK-NEXT:    eon w12, w10, w9, lsl #31
; CHECK-NEXT:    str w11, [x8]
; CHECK-NEXT:    bic w11, w10, w9, asr #10
; CHECK-NEXT:    str w12, [x8]
; CHECK-NEXT:    eor w12, w10, w9, asr #10
; CHECK-NEXT:    str w11, [x8]
; CHECK-NEXT:    orn w11, w10, w9, lsr #1
; CHECK-NEXT:    str w12, [x8]
; CHECK-NEXT:    eor w12, w10, w9, lsr #1
; CHECK-NEXT:    str w11, [x8]
; CHECK-NEXT:    eon w11, w10, w9, ror #20
; CHECK-NEXT:    and w9, w10, w9, ror #20
; CHECK-NEXT:    str w12, [x8]
; CHECK-NEXT:    str w11, [x8]
; CHECK-NEXT:    str w9, [x8]
; CHECK-NEXT:    ret
  %val1 = load i32, ptr @var1_32
  %val2 = load i32, ptr @var2_32

  ; First check basic and/bic/or/orn/eor/eon patterns with no shift
  %neg_val2 = xor i32 -1, %val2

  %and_noshift = and i32 %val1, %val2
  store volatile i32 %and_noshift, ptr @var1_32
  %bic_noshift = and i32 %neg_val2, %val1
  store volatile i32 %bic_noshift, ptr @var1_32

  %or_noshift = or i32 %val1, %val2
  store volatile i32 %or_noshift, ptr @var1_32
  %orn_noshift = or i32 %neg_val2, %val1
  store volatile i32 %orn_noshift, ptr @var1_32

  %xor_noshift = xor i32 %val1, %val2
  store volatile i32 %xor_noshift, ptr @var1_32
  %xorn_noshift = xor i32 %neg_val2, %val1
  store volatile i32 %xorn_noshift, ptr @var1_32

  ; Check the maximum shift on each
  %operand_lsl31 = shl i32 %val2, 31
  %neg_operand_lsl31 = xor i32 -1, %operand_lsl31

  %and_lsl31 = and i32 %val1, %operand_lsl31
  store volatile i32 %and_lsl31, ptr @var1_32
  %bic_lsl31 = and i32 %val1, %neg_operand_lsl31
  store volatile i32 %bic_lsl31, ptr @var1_32

  %or_lsl31 = or i32 %val1, %operand_lsl31
  store volatile i32 %or_lsl31, ptr @var1_32
  %orn_lsl31 = or i32 %val1, %neg_operand_lsl31
  store volatile i32 %orn_lsl31, ptr @var1_32

  %xor_lsl31 = xor i32 %val1, %operand_lsl31
  store volatile i32 %xor_lsl31, ptr @var1_32
  %xorn_lsl31 = xor i32 %val1, %neg_operand_lsl31
  store volatile i32 %xorn_lsl31, ptr @var1_32

  ; Check other shifts on a subset
  %operand_asr10 = ashr i32 %val2, 10
  %neg_operand_asr10 = xor i32 -1, %operand_asr10

  %bic_asr10 = and i32 %val1, %neg_operand_asr10
  store volatile i32 %bic_asr10, ptr @var1_32
  %xor_asr10 = xor i32 %val1, %operand_asr10
  store volatile i32 %xor_asr10, ptr @var1_32

  %operand_lsr1 = lshr i32 %val2, 1
  %neg_operand_lsr1 = xor i32 -1, %operand_lsr1

  %orn_lsr1 = or i32 %val1, %neg_operand_lsr1
  store volatile i32 %orn_lsr1, ptr @var1_32
  %xor_lsr1 = xor i32 %val1, %operand_lsr1
  store volatile i32 %xor_lsr1, ptr @var1_32

  %operand_ror20_big = shl i32 %val2, 12
  %operand_ror20_small = lshr i32 %val2, 20
  %operand_ror20 = or i32 %operand_ror20_big, %operand_ror20_small
  %neg_operand_ror20 = xor i32 -1, %operand_ror20

  %xorn_ror20 = xor i32 %val1, %neg_operand_ror20
  store volatile i32 %xorn_ror20, ptr @var1_32
  %and_ror20 = and i32 %val1, %operand_ror20
  store volatile i32 %and_ror20, ptr @var1_32

  ret void
}

define void @logical_64bit() minsize {
; CHECK-LABEL: logical_64bit:
; CHECK:       // %bb.0:
; CHECK-NEXT:    adrp x8, :got:var1_64
; CHECK-NEXT:    adrp x9, :got:var2_64
; CHECK-NEXT:    ldr x8, [x8, :got_lo12:var1_64]
; CHECK-NEXT:    ldr x9, [x9, :got_lo12:var2_64]
; CHECK-NEXT:    ldr x10, [x8]
; CHECK-NEXT:    ldr x9, [x9]
; CHECK-NEXT:    and x11, x10, x9
; CHECK-NEXT:    bic x12, x10, x9
; CHECK-NEXT:    str x11, [x8]
; CHECK-NEXT:    orr x11, x10, x9
; CHECK-NEXT:    str x12, [x8]
; CHECK-NEXT:    orn x12, x10, x9
; CHECK-NEXT:    str x11, [x8]
; CHECK-NEXT:    eor x11, x10, x9
; CHECK-NEXT:    str x12, [x8]
; CHECK-NEXT:    eon x12, x9, x10
; CHECK-NEXT:    str x11, [x8]
; CHECK-NEXT:    and x11, x10, x9, lsl #63
; CHECK-NEXT:    str x12, [x8]
; CHECK-NEXT:    bic x12, x10, x9, lsl #63
; CHECK-NEXT:    str x11, [x8]
; CHECK-NEXT:    orr x11, x10, x9, lsl #63
; CHECK-NEXT:    str x12, [x8]
; CHECK-NEXT:    orn x12, x10, x9, lsl #63
; CHECK-NEXT:    str x11, [x8]
; CHECK-NEXT:    eor x11, x10, x9, lsl #63
; CHECK-NEXT:    str x12, [x8]
; CHECK-NEXT:    eon x12, x10, x9, lsl #63
; CHECK-NEXT:    str x11, [x8]
; CHECK-NEXT:    bic x11, x10, x9, asr #10
; CHECK-NEXT:    str x12, [x8]
; CHECK-NEXT:    eor x12, x10, x9, asr #10
; CHECK-NEXT:    str x11, [x8]
; CHECK-NEXT:    orn x11, x10, x9, lsr #1
; CHECK-NEXT:    str x12, [x8]
; CHECK-NEXT:    eor x12, x10, x9, lsr #1
; CHECK-NEXT:    str x11, [x8]
; CHECK-NEXT:    eon x11, x10, x9, ror #20
; CHECK-NEXT:    and x9, x10, x9, ror #20
; CHECK-NEXT:    str x12, [x8]
; CHECK-NEXT:    str x11, [x8]
; CHECK-NEXT:    str x9, [x8]
; CHECK-NEXT:    ret
  %val1 = load i64, ptr @var1_64
  %val2 = load i64, ptr @var2_64

  ; First check basic and/bic/or/orn/eor/eon patterns with no shift
  %neg_val2 = xor i64 -1, %val2

  %and_noshift = and i64 %val1, %val2
  store volatile i64 %and_noshift, ptr @var1_64
  %bic_noshift = and i64 %neg_val2, %val1
  store volatile i64 %bic_noshift, ptr @var1_64

  %or_noshift = or i64 %val1, %val2
  store volatile i64 %or_noshift, ptr @var1_64
  %orn_noshift = or i64 %neg_val2, %val1
  store volatile i64 %orn_noshift, ptr @var1_64

  %xor_noshift = xor i64 %val1, %val2
  store volatile i64 %xor_noshift, ptr @var1_64
  %xorn_noshift = xor i64 %neg_val2, %val1
  store volatile i64 %xorn_noshift, ptr @var1_64

  ; Check the maximum shift on each
  %operand_lsl63 = shl i64 %val2, 63
  %neg_operand_lsl63 = xor i64 -1, %operand_lsl63

  %and_lsl63 = and i64 %val1, %operand_lsl63
  store volatile i64 %and_lsl63, ptr @var1_64
  %bic_lsl63 = and i64 %val1, %neg_operand_lsl63
  store volatile i64 %bic_lsl63, ptr @var1_64

  %or_lsl63 = or i64 %val1, %operand_lsl63
  store volatile i64 %or_lsl63, ptr @var1_64
  %orn_lsl63 = or i64 %val1, %neg_operand_lsl63
  store volatile i64 %orn_lsl63, ptr @var1_64

  %xor_lsl63 = xor i64 %val1, %operand_lsl63
  store volatile i64 %xor_lsl63, ptr @var1_64
  %xorn_lsl63 = xor i64 %val1, %neg_operand_lsl63
  store volatile i64 %xorn_lsl63, ptr @var1_64

  ; Check other shifts on a subset
  %operand_asr10 = ashr i64 %val2, 10
  %neg_operand_asr10 = xor i64 -1, %operand_asr10

  %bic_asr10 = and i64 %val1, %neg_operand_asr10
  store volatile i64 %bic_asr10, ptr @var1_64
  %xor_asr10 = xor i64 %val1, %operand_asr10
  store volatile i64 %xor_asr10, ptr @var1_64

  %operand_lsr1 = lshr i64 %val2, 1
  %neg_operand_lsr1 = xor i64 -1, %operand_lsr1

  %orn_lsr1 = or i64 %val1, %neg_operand_lsr1
  store volatile i64 %orn_lsr1, ptr @var1_64
  %xor_lsr1 = xor i64 %val1, %operand_lsr1
  store volatile i64 %xor_lsr1, ptr @var1_64

  ; Construct a rotate-right from a bunch of other logical
  ; operations. DAGCombiner should ensure we the ROTR during
  ; selection
  %operand_ror20_big = shl i64 %val2, 44
  %operand_ror20_small = lshr i64 %val2, 20
  %operand_ror20 = or i64 %operand_ror20_big, %operand_ror20_small
  %neg_operand_ror20 = xor i64 -1, %operand_ror20

  %xorn_ror20 = xor i64 %val1, %neg_operand_ror20
  store volatile i64 %xorn_ror20, ptr @var1_64
  %and_ror20 = and i64 %val1, %operand_ror20
  store volatile i64 %and_ror20, ptr @var1_64

  ret void
}

define void @flag_setting() {
; CHECK-LABEL: flag_setting:
; CHECK:       // %bb.0:
; CHECK-NEXT:    adrp x8, :got:var1_64
; CHECK-NEXT:    adrp x10, :got:var2_64
; CHECK-NEXT:    ldr x8, [x8, :got_lo12:var1_64]
; CHECK-NEXT:    ldr x10, [x10, :got_lo12:var2_64]
; CHECK-NEXT:    ldr x9, [x8]
; CHECK-NEXT:    ldr x10, [x10]
; CHECK-NEXT:    tst x9, x10
; CHECK-NEXT:    b.gt .LBB2_4
; CHECK-NEXT:  // %bb.1: // %test2
; CHECK-NEXT:    tst x9, x10, lsl #63
; CHECK-NEXT:    b.lt .LBB2_4
; CHECK-NEXT:  // %bb.2: // %test3
; CHECK-NEXT:    and x10, x9, x10, asr #12
; CHECK-NEXT:    cmp x10, #1
; CHECK-NEXT:    b.ge .LBB2_4
; CHECK-NEXT:  // %bb.3: // %other_exit
; CHECK-NEXT:    str x9, [x8]
; CHECK-NEXT:  .LBB2_4: // %common.ret
; CHECK-NEXT:    ret
  %val1 = load i64, ptr @var1_64
  %val2 = load i64, ptr @var2_64

  %simple_and = and i64 %val1, %val2
  %tst1 = icmp sgt i64 %simple_and, 0
  br i1 %tst1, label %ret, label %test2, !prof !1

test2:
  %shifted_op = shl i64 %val2, 63
  %shifted_and = and i64 %val1, %shifted_op
  %tst2 = icmp slt i64 %shifted_and, 0
  br i1 %tst2, label %ret, label %test3, !prof !1

test3:
  %asr_op = ashr i64 %val2, 12
  %asr_and = and i64 %asr_op, %val1
  %tst3 = icmp sgt i64 %asr_and, 0
  br i1 %tst3, label %ret, label %other_exit, !prof !1

other_exit:
  store volatile i64 %val1, ptr @var1_64
  ret void
ret:
  ret void
}

define i64 @add_swap_rhs_lhs_i64(i64 %0, i64 %1) {
; CHECK-LABEL: add_swap_rhs_lhs_i64:
; CHECK:       // %bb.0:
; CHECK-NEXT:    lsl x8, x0, #8
; CHECK-NEXT:    add x0, x8, x1, lsl #3
; CHECK-NEXT:    ret
  %3 = shl i64 %0, 8
  %4 = shl i64 %1, 3
  %5 = add i64 %4, %3
  ret i64 %5
}

define i64 @add_swap_no_op_i64(i64 %0, i64 %1, ptr %2) {
; CHECK-LABEL: add_swap_no_op_i64:
; CHECK:       // %bb.0:
; CHECK-NEXT:    lsl x8, x1, #3
; CHECK-NEXT:    add x0, x8, x0, lsl #8
; CHECK-NEXT:    str x8, [x2]
; CHECK-NEXT:    ret
  %4 = shl i64 %0, 8
  %5 = shl i64 %1, 3
  store i64 %5, ptr %2
  %6 = add i64 %5, %4
  ret i64 %6
}

define i32 @add_swap_rhs_lhs_i32(i32 %0, i32 %1) {
; CHECK-LABEL: add_swap_rhs_lhs_i32:
; CHECK:       // %bb.0:
; CHECK-NEXT:    lsl w8, w0, #8
; CHECK-NEXT:    add w0, w8, w1, lsl #3
; CHECK-NEXT:    ret
  %3 = shl i32 %0, 8
  %4 = shl i32 %1, 3
  %5 = add i32 %4, %3
  ret i32 %5
}

define i32 @add_swap_no_op_i32(i32 %0, i32 %1, ptr %2) {
; CHECK-LABEL: add_swap_no_op_i32:
; CHECK:       // %bb.0:
; CHECK-NEXT:    lsl w8, w1, #3
; CHECK-NEXT:    add w0, w8, w0, lsl #8
; CHECK-NEXT:    str w8, [x2]
; CHECK-NEXT:    ret
  %4 = shl i32 %0, 8
  %5 = shl i32 %1, 3
  store i32 %5, ptr %2
  %6 = add i32 %5, %4
  ret i32 %6
}

!1 = !{!"branch_weights", i32 1, i32 1}