llvm/llvm/test/CodeGen/AArch64/srem-pow2.ll

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

define i16 @fold_srem_1_i16(i16 %x) {
; CHECK-LABEL: fold_srem_1_i16:
; CHECK:       // %bb.0:
; CHECK-NEXT:    mov w0, wzr
; CHECK-NEXT:    ret
  %1 = srem i16 %x, 1
  ret i16 %1
}

define i32 @fold_srem_1_i32(i32 %x) {
; CHECK-LABEL: fold_srem_1_i32:
; CHECK:       // %bb.0:
; CHECK-NEXT:    mov w0, wzr
; CHECK-NEXT:    ret
  %1 = srem i32 %x, 1
  ret i32 %1
}

define i64 @fold_srem_1_i64(i64 %x) {
; CHECK-LABEL: fold_srem_1_i64:
; CHECK:       // %bb.0:
; CHECK-NEXT:    mov x0, xzr
; CHECK-NEXT:    ret
  %1 = srem i64 %x, 1
  ret i64 %1
}

define i16 @fold_srem_2_i16(i16 %x) {
; CHECK-LABEL: fold_srem_2_i16:
; CHECK:       // %bb.0:
; CHECK-NEXT:    and w8, w0, #0x8000
; CHECK-NEXT:    add w8, w0, w8, lsr #15
; CHECK-NEXT:    and w8, w8, #0xfffffffe
; CHECK-NEXT:    sub w0, w0, w8
; CHECK-NEXT:    ret
  %1 = srem i16 %x, 2
  ret i16 %1
}

define i32 @fold_srem_2_i64(i32 %x) {
; CHECK-LABEL: fold_srem_2_i64:
; CHECK:       // %bb.0:
; CHECK-NEXT:    and w8, w0, #0x1
; CHECK-NEXT:    cmp w0, #0
; CHECK-NEXT:    cneg w0, w8, lt
; CHECK-NEXT:    ret
  %1 = srem i32 %x, 2
  ret i32 %1
}

define i64 @fold_srem_2_i32(i64 %x) {
; CHECK-LABEL: fold_srem_2_i32:
; CHECK:       // %bb.0:
; CHECK-NEXT:    and x8, x0, #0x1
; CHECK-NEXT:    cmp x0, #0
; CHECK-NEXT:    cneg x0, x8, lt
; CHECK-NEXT:    ret
  %1 = srem i64 %x, 2
  ret i64 %1
}

define i16 @fold_srem_pow2_i16(i16 %x) {
; CHECK-LABEL: fold_srem_pow2_i16:
; CHECK:       // %bb.0:
; CHECK-NEXT:    sxth w8, w0
; CHECK-NEXT:    ubfx w8, w8, #25, #6
; CHECK-NEXT:    add w8, w0, w8
; CHECK-NEXT:    and w8, w8, #0xffffffc0
; CHECK-NEXT:    sub w0, w0, w8
; CHECK-NEXT:    ret
  %1 = srem i16 %x, 64
  ret i16 %1
}

define i32 @fold_srem_pow2_i32(i32 %x) {
; CHECK-LABEL: fold_srem_pow2_i32:
; CHECK:       // %bb.0:
; CHECK-NEXT:    negs w8, w0
; CHECK-NEXT:    and w9, w0, #0x3f
; CHECK-NEXT:    and w8, w8, #0x3f
; CHECK-NEXT:    csneg w0, w9, w8, mi
; CHECK-NEXT:    ret
  %1 = srem i32 %x, 64
  ret i32 %1
}

define i64 @fold_srem_pow2_i64(i64 %x) {
; CHECK-LABEL: fold_srem_pow2_i64:
; CHECK:       // %bb.0:
; CHECK-NEXT:    negs x8, x0
; CHECK-NEXT:    and x9, x0, #0x3f
; CHECK-NEXT:    and x8, x8, #0x3f
; CHECK-NEXT:    csneg x0, x9, x8, mi
; CHECK-NEXT:    ret
  %1 = srem i64 %x, 64
  ret i64 %1
}

define i16 @fold_srem_smax_i16(i16 %x) {
; CHECK-LABEL: fold_srem_smax_i16:
; CHECK:       // %bb.0:
; CHECK-NEXT:    sxth w8, w0
; CHECK-NEXT:    ubfx w8, w8, #16, #15
; CHECK-NEXT:    add w8, w0, w8
; CHECK-NEXT:    and w8, w8, #0xffff8000
; CHECK-NEXT:    add w0, w0, w8
; CHECK-NEXT:    ret
  %1 = srem i16 %x, 32768
  ret i16 %1
}

define i32 @fold_srem_smax_i32(i32 %x) {
; CHECK-LABEL: fold_srem_smax_i32:
; CHECK:       // %bb.0:
; CHECK-NEXT:    negs w8, w0
; CHECK-NEXT:    and w9, w0, #0x7fffffff
; CHECK-NEXT:    and w8, w8, #0x7fffffff
; CHECK-NEXT:    csneg w0, w9, w8, mi
; CHECK-NEXT:    ret
  %1 = srem i32 %x, 2147483648
  ret i32 %1
}

define i64 @fold_srem_smax_i64(i64 %x) {
; CHECK-LABEL: fold_srem_smax_i64:
; CHECK:       // %bb.0:
; CHECK-NEXT:    negs x8, x0
; CHECK-NEXT:    and x9, x0, #0x7fffffffffffffff
; CHECK-NEXT:    and x8, x8, #0x7fffffffffffffff
; CHECK-NEXT:    csneg x0, x9, x8, mi
; CHECK-NEXT:    ret
  %1 = srem i64 %x, -9223372036854775808
  ret i64 %1
}