150 lines
4.1 KiB
LLVM
150 lines
4.1 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
|
|
; RUN: llc -mtriple=aarch64-unknown-linux-gnu < %s | FileCheck %s
|
|
|
|
define i32 @fold_srem_positive_odd(i32 %x) {
|
|
; CHECK-LABEL: fold_srem_positive_odd:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: mov w8, #37253
|
|
; CHECK-NEXT: movk w8, #44150, lsl #16
|
|
; CHECK-NEXT: smull x8, w0, w8
|
|
; CHECK-NEXT: lsr x8, x8, #32
|
|
; CHECK-NEXT: add w8, w8, w0
|
|
; CHECK-NEXT: asr w9, w8, #6
|
|
; CHECK-NEXT: add w8, w9, w8, lsr #31
|
|
; CHECK-NEXT: mov w9, #95
|
|
; CHECK-NEXT: msub w0, w8, w9, w0
|
|
; CHECK-NEXT: ret
|
|
%1 = srem i32 %x, 95
|
|
ret i32 %1
|
|
}
|
|
|
|
|
|
define i32 @fold_srem_positive_even(i32 %x) {
|
|
; CHECK-LABEL: fold_srem_positive_even:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: mov w8, #36849
|
|
; CHECK-NEXT: movk w8, #15827, lsl #16
|
|
; CHECK-NEXT: smull x8, w0, w8
|
|
; CHECK-NEXT: lsr x9, x8, #63
|
|
; CHECK-NEXT: asr x8, x8, #40
|
|
; CHECK-NEXT: add w8, w8, w9
|
|
; CHECK-NEXT: mov w9, #1060
|
|
; CHECK-NEXT: msub w0, w8, w9, w0
|
|
; CHECK-NEXT: ret
|
|
%1 = srem i32 %x, 1060
|
|
ret i32 %1
|
|
}
|
|
|
|
|
|
define i32 @fold_srem_negative_odd(i32 %x) {
|
|
; CHECK-LABEL: fold_srem_negative_odd:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: mov w8, #65445
|
|
; CHECK-NEXT: movk w8, #42330, lsl #16
|
|
; CHECK-NEXT: smull x8, w0, w8
|
|
; CHECK-NEXT: lsr x9, x8, #63
|
|
; CHECK-NEXT: asr x8, x8, #40
|
|
; CHECK-NEXT: add w8, w8, w9
|
|
; CHECK-NEXT: mov w9, #-723
|
|
; CHECK-NEXT: msub w0, w8, w9, w0
|
|
; CHECK-NEXT: ret
|
|
%1 = srem i32 %x, -723
|
|
ret i32 %1
|
|
}
|
|
|
|
|
|
define i32 @fold_srem_negative_even(i32 %x) {
|
|
; CHECK-LABEL: fold_srem_negative_even:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: mov w8, #62439
|
|
; CHECK-NEXT: movk w8, #64805, lsl #16
|
|
; CHECK-NEXT: smull x8, w0, w8
|
|
; CHECK-NEXT: lsr x9, x8, #63
|
|
; CHECK-NEXT: asr x8, x8, #40
|
|
; CHECK-NEXT: add w8, w8, w9
|
|
; CHECK-NEXT: mov w9, #-22981
|
|
; CHECK-NEXT: msub w0, w8, w9, w0
|
|
; CHECK-NEXT: ret
|
|
%1 = srem i32 %x, -22981
|
|
ret i32 %1
|
|
}
|
|
|
|
|
|
; Don't fold if we can combine srem with sdiv.
|
|
define i32 @combine_srem_sdiv(i32 %x) {
|
|
; CHECK-LABEL: combine_srem_sdiv:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: mov w8, #37253
|
|
; CHECK-NEXT: movk w8, #44150, lsl #16
|
|
; CHECK-NEXT: smull x8, w0, w8
|
|
; CHECK-NEXT: lsr x8, x8, #32
|
|
; CHECK-NEXT: add w8, w8, w0
|
|
; CHECK-NEXT: asr w9, w8, #6
|
|
; CHECK-NEXT: add w8, w9, w8, lsr #31
|
|
; CHECK-NEXT: mov w9, #95
|
|
; CHECK-NEXT: msub w9, w8, w9, w0
|
|
; CHECK-NEXT: add w0, w9, w8
|
|
; CHECK-NEXT: ret
|
|
%1 = srem i32 %x, 95
|
|
%2 = sdiv i32 %x, 95
|
|
%3 = add i32 %1, %2
|
|
ret i32 %3
|
|
}
|
|
|
|
; Don't fold for divisors that are a power of two.
|
|
define i32 @dont_fold_srem_power_of_two(i32 %x) {
|
|
; CHECK-LABEL: dont_fold_srem_power_of_two:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: add w8, w0, #63 // =63
|
|
; CHECK-NEXT: cmp w0, #0 // =0
|
|
; CHECK-NEXT: csel w8, w8, w0, lt
|
|
; CHECK-NEXT: and w8, w8, #0xffffffc0
|
|
; CHECK-NEXT: sub w0, w0, w8
|
|
; CHECK-NEXT: ret
|
|
%1 = srem i32 %x, 64
|
|
ret i32 %1
|
|
}
|
|
|
|
; Don't fold if the divisor is one.
|
|
define i32 @dont_fold_srem_one(i32 %x) {
|
|
; CHECK-LABEL: dont_fold_srem_one:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: mov w0, wzr
|
|
; CHECK-NEXT: ret
|
|
%1 = srem i32 %x, 1
|
|
ret i32 %1
|
|
}
|
|
|
|
; Don't fold if the divisor is 2^31.
|
|
define i32 @dont_fold_srem_i32_smax(i32 %x) {
|
|
; CHECK-LABEL: dont_fold_srem_i32_smax:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: mov w8, #2147483647
|
|
; CHECK-NEXT: add w8, w0, w8
|
|
; CHECK-NEXT: cmp w0, #0 // =0
|
|
; CHECK-NEXT: csel w8, w8, w0, lt
|
|
; CHECK-NEXT: and w8, w8, #0x80000000
|
|
; CHECK-NEXT: add w0, w0, w8
|
|
; CHECK-NEXT: ret
|
|
%1 = srem i32 %x, 2147483648
|
|
ret i32 %1
|
|
}
|
|
|
|
; Don't fold i64 srem
|
|
define i64 @dont_fold_srem_i64(i64 %x) {
|
|
; CHECK-LABEL: dont_fold_srem_i64:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: mov x8, #58849
|
|
; CHECK-NEXT: movk x8, #48148, lsl #16
|
|
; CHECK-NEXT: movk x8, #33436, lsl #32
|
|
; CHECK-NEXT: movk x8, #21399, lsl #48
|
|
; CHECK-NEXT: smulh x8, x0, x8
|
|
; CHECK-NEXT: asr x9, x8, #5
|
|
; CHECK-NEXT: add x8, x9, x8, lsr #63
|
|
; CHECK-NEXT: mov w9, #98
|
|
; CHECK-NEXT: msub x0, x8, x9, x0
|
|
; CHECK-NEXT: ret
|
|
%1 = srem i64 %x, 98
|
|
ret i64 %1
|
|
}
|