test/CodeGen/AArch64/urem-lkk.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -mtriple=aarch64-unknown-linux-gnu < %s | FileCheck %s
   3
   4 define i32 @fold_urem_positive_odd(i32 %x) {
   5 ; CHECK-LABEL: fold_urem_positive_odd:
   6 ; CHECK:       // %bb.0:
   7 ; CHECK-NEXT:    mov w8, #8969
   8 ; CHECK-NEXT:    movk w8, #22765, lsl #16
   9 ; CHECK-NEXT:    umull x8, w0, w8
  10 ; CHECK-NEXT:    lsr x8, x8, #32
  11 ; CHECK-NEXT:    sub w9, w0, w8
  12 ; CHECK-NEXT:    add w8, w8, w9, lsr #1
  13 ; CHECK-NEXT:    lsr w8, w8, #6
  14 ; CHECK-NEXT:    mov w9, #95
  15 ; CHECK-NEXT:    msub w0, w8, w9, w0
  16 ; CHECK-NEXT:    ret
  17   %1 = urem i32 %x, 95
  18   ret i32 %1
  19 }
  20
  21
  22 define i32 @fold_urem_positive_even(i32 %x) {
  23 ; CHECK-LABEL: fold_urem_positive_even:
  24 ; CHECK:       // %bb.0:
  25 ; CHECK-NEXT:    mov w8, #16323
  26 ; CHECK-NEXT:    movk w8, #63310, lsl #16
  27 ; CHECK-NEXT:    umull x8, w0, w8
  28 ; CHECK-NEXT:    lsr x8, x8, #42
  29 ; CHECK-NEXT:    mov w9, #1060
  30 ; CHECK-NEXT:    msub w0, w8, w9, w0
  31 ; CHECK-NEXT:    ret
  32   %1 = urem i32 %x, 1060
  33   ret i32 %1
  34 }
  35
  36
  37 ; Don't fold if we can combine urem with udiv.
  38 define i32 @combine_urem_udiv(i32 %x) {
  39 ; CHECK-LABEL: combine_urem_udiv:
  40 ; CHECK:       // %bb.0:
  41 ; CHECK-NEXT:    mov w8, #8969
  42 ; CHECK-NEXT:    movk w8, #22765, lsl #16
  43 ; CHECK-NEXT:    umull x8, w0, w8
  44 ; CHECK-NEXT:    lsr x8, x8, #32
  45 ; CHECK-NEXT:    sub w9, w0, w8
  46 ; CHECK-NEXT:    add w8, w8, w9, lsr #1
  47 ; CHECK-NEXT:    lsr w8, w8, #6
  48 ; CHECK-NEXT:    mov w9, #95
  49 ; CHECK-NEXT:    msub w9, w8, w9, w0
  50 ; CHECK-NEXT:    add w0, w9, w8
  51 ; CHECK-NEXT:    ret
  52   %1 = urem i32 %x, 95
  53   %2 = udiv i32 %x, 95
  54   %3 = add i32 %1, %2
  55   ret i32 %3
  56 }
  57
  58 ; Don't fold for divisors that are a power of two.
  59 define i32 @dont_fold_urem_power_of_two(i32 %x) {
  60 ; CHECK-LABEL: dont_fold_urem_power_of_two:
  61 ; CHECK:       // %bb.0:
  62 ; CHECK-NEXT:    and w0, w0, #0x3f
  63 ; CHECK-NEXT:    ret
  64   %1 = urem i32 %x, 64
  65   ret i32 %1
  66 }
  67
  68 ; Don't fold if the divisor is one.
  69 define i32 @dont_fold_urem_one(i32 %x) {
  70 ; CHECK-LABEL: dont_fold_urem_one:
  71 ; CHECK:       // %bb.0:
  72 ; CHECK-NEXT:    mov w0, wzr
  73 ; CHECK-NEXT:    ret
  74   %1 = urem i32 %x, 1
  75   ret i32 %1
  76 }
  77
  78 ; Don't fold if the divisor is 2^32.
  79 define i32 @dont_fold_urem_i32_umax(i32 %x) {
  80 ; CHECK-LABEL: dont_fold_urem_i32_umax:
  81 ; CHECK:       // %bb.0:
  82 ; CHECK-NEXT:    ret
  83   %1 = urem i32 %x, 4294967296
  84   ret i32 %1
  85 }
  86
  87 ; Don't fold i64 urem
  88 define i64 @dont_fold_urem_i64(i64 %x) {
  89 ; CHECK-LABEL: dont_fold_urem_i64:
  90 ; CHECK:       // %bb.0:
  91 ; CHECK-NEXT:    mov x9, #58849
  92 ; CHECK-NEXT:    movk x9, #48148, lsl #16
  93 ; CHECK-NEXT:    movk x9, #33436, lsl #32
  94 ; CHECK-NEXT:    lsr x8, x0, #1
  95 ; CHECK-NEXT:    movk x9, #21399, lsl #48
  96 ; CHECK-NEXT:    umulh x8, x8, x9
  97 ; CHECK-NEXT:    lsr x8, x8, #4
  98 ; CHECK-NEXT:    mov w9, #98
  99 ; CHECK-NEXT:    msub x0, x8, x9, x0
 100 ; CHECK-NEXT:    ret
 101   %1 = urem i64 %x, 98
 102   ret i64 %1
 103 }