xref: /aosp_15_r20/external/llvm/test/CodeGen/AArch64/arm64-long-shift.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker; RUN: llc < %s -march=arm64 -mcpu=cyclone | FileCheck %s
2*9880d681SAndroid Build Coastguard Worker
3*9880d681SAndroid Build Coastguard Workerdefine i128 @shl(i128 %r, i128 %s) nounwind readnone {
4*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: shl:
5*9880d681SAndroid Build Coastguard Worker; CHECK: orr w[[SIXTY_FOUR:[0-9]+]], wzr, #0x40
6*9880d681SAndroid Build Coastguard Worker; CHECK: sub [[REV_SHIFT:x[0-9]+]], x[[SIXTY_FOUR]], x2
7*9880d681SAndroid Build Coastguard Worker; CHECK: lsr  [[LO_FOR_HI_NORMAL:x[0-9]+]], x0, [[REV_SHIFT]]
8*9880d681SAndroid Build Coastguard Worker; CHECK: cmp x2, #0
9*9880d681SAndroid Build Coastguard Worker; CHECK: csel [[LO_FOR_HI:x[0-9]+]], xzr, [[LO_FOR_HI_NORMAL]], eq
10*9880d681SAndroid Build Coastguard Worker; CHECK: lsl  [[HI_FOR_HI:x[0-9]+]], x1, x2
11*9880d681SAndroid Build Coastguard Worker; CHECK: orr [[HI_NORMAL:x[0-9]+]], [[LO_FOR_HI]], [[HI_FOR_HI]]
12*9880d681SAndroid Build Coastguard Worker; CHECK: sub [[EXTRA_SHIFT:x[0-9]+]], x2, #64
13*9880d681SAndroid Build Coastguard Worker; CHECK: lsl  [[HI_BIG_SHIFT:x[0-9]+]], x0, [[EXTRA_SHIFT]]
14*9880d681SAndroid Build Coastguard Worker; CHECK: cmp   [[EXTRA_SHIFT]], #0
15*9880d681SAndroid Build Coastguard Worker; CHECK: csel  x1, [[HI_BIG_SHIFT]], [[HI_NORMAL]], ge
16*9880d681SAndroid Build Coastguard Worker; CHECK: lsl  [[SMALLSHIFT_LO:x[0-9]+]], x0, x2
17*9880d681SAndroid Build Coastguard Worker; CHECK: csel  x0, xzr, [[SMALLSHIFT_LO]], ge
18*9880d681SAndroid Build Coastguard Worker; CHECK: ret
19*9880d681SAndroid Build Coastguard Worker
20*9880d681SAndroid Build Coastguard Worker  %shl = shl i128 %r, %s
21*9880d681SAndroid Build Coastguard Worker  ret i128 %shl
22*9880d681SAndroid Build Coastguard Worker}
23*9880d681SAndroid Build Coastguard Worker
24*9880d681SAndroid Build Coastguard Workerdefine i128 @ashr(i128 %r, i128 %s) nounwind readnone {
25*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: ashr:
26*9880d681SAndroid Build Coastguard Worker; CHECK: orr w[[SIXTY_FOUR:[0-9]+]], wzr, #0x40
27*9880d681SAndroid Build Coastguard Worker; CHECK: sub [[REV_SHIFT:x[0-9]+]], x[[SIXTY_FOUR]], x2
28*9880d681SAndroid Build Coastguard Worker; CHECK: lsl  [[HI_FOR_LO_NORMAL:x[0-9]+]], x1, [[REV_SHIFT]]
29*9880d681SAndroid Build Coastguard Worker; CHECK: cmp x2, #0
30*9880d681SAndroid Build Coastguard Worker; CHECK: csel [[HI_FOR_LO:x[0-9]+]], xzr, [[HI_FOR_LO_NORMAL]], eq
31*9880d681SAndroid Build Coastguard Worker; CHECK: lsr  [[LO_FOR_LO:x[0-9]+]], x0, x2
32*9880d681SAndroid Build Coastguard Worker; CHECK: orr [[LO_NORMAL:x[0-9]+]], [[LO_FOR_LO]], [[HI_FOR_LO]]
33*9880d681SAndroid Build Coastguard Worker; CHECK: sub [[EXTRA_SHIFT:x[0-9]+]], x2, #64
34*9880d681SAndroid Build Coastguard Worker; CHECK: asr  [[LO_BIG_SHIFT:x[0-9]+]], x1, [[EXTRA_SHIFT]]
35*9880d681SAndroid Build Coastguard Worker; CHECK: cmp   [[EXTRA_SHIFT]], #0
36*9880d681SAndroid Build Coastguard Worker; CHECK: csel  x0, [[LO_BIG_SHIFT]], [[LO_NORMAL]], ge
37*9880d681SAndroid Build Coastguard Worker; CHECK: asr  [[SMALLSHIFT_HI:x[0-9]+]], x1, x2
38*9880d681SAndroid Build Coastguard Worker; CHECK: asr [[BIGSHIFT_HI:x[0-9]+]], x1, #63
39*9880d681SAndroid Build Coastguard Worker; CHECK: csel x1, [[BIGSHIFT_HI]], [[SMALLSHIFT_HI]], ge
40*9880d681SAndroid Build Coastguard Worker; CHECK: ret
41*9880d681SAndroid Build Coastguard Worker
42*9880d681SAndroid Build Coastguard Worker  %shr = ashr i128 %r, %s
43*9880d681SAndroid Build Coastguard Worker  ret i128 %shr
44*9880d681SAndroid Build Coastguard Worker}
45*9880d681SAndroid Build Coastguard Worker
46*9880d681SAndroid Build Coastguard Workerdefine i128 @lshr(i128 %r, i128 %s) nounwind readnone {
47*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: lshr:
48*9880d681SAndroid Build Coastguard Worker; CHECK: orr w[[SIXTY_FOUR:[0-9]+]], wzr, #0x40
49*9880d681SAndroid Build Coastguard Worker; CHECK: sub [[REV_SHIFT:x[0-9]+]], x[[SIXTY_FOUR]], x2
50*9880d681SAndroid Build Coastguard Worker; CHECK: lsl  [[HI_FOR_LO_NORMAL:x[0-9]+]], x1, [[REV_SHIFT]]
51*9880d681SAndroid Build Coastguard Worker; CHECK: cmp x2, #0
52*9880d681SAndroid Build Coastguard Worker; CHECK: csel [[HI_FOR_LO:x[0-9]+]], xzr, [[HI_FOR_LO_NORMAL]], eq
53*9880d681SAndroid Build Coastguard Worker; CHECK: lsr  [[LO_FOR_LO:x[0-9]+]], x0, x2
54*9880d681SAndroid Build Coastguard Worker; CHECK: orr [[LO_NORMAL:x[0-9]+]], [[LO_FOR_LO]], [[HI_FOR_LO]]
55*9880d681SAndroid Build Coastguard Worker; CHECK: sub [[EXTRA_SHIFT:x[0-9]+]], x2, #64
56*9880d681SAndroid Build Coastguard Worker; CHECK: lsr  [[LO_BIG_SHIFT:x[0-9]+]], x1, [[EXTRA_SHIFT]]
57*9880d681SAndroid Build Coastguard Worker; CHECK: cmp   [[EXTRA_SHIFT]], #0
58*9880d681SAndroid Build Coastguard Worker; CHECK: csel  x0, [[LO_BIG_SHIFT]], [[LO_NORMAL]], ge
59*9880d681SAndroid Build Coastguard Worker; CHECK: lsr  [[SMALLSHIFT_HI:x[0-9]+]], x1, x2
60*9880d681SAndroid Build Coastguard Worker; CHECK: csel x1, xzr, [[SMALLSHIFT_HI]], ge
61*9880d681SAndroid Build Coastguard Worker; CHECK: ret
62*9880d681SAndroid Build Coastguard Worker
63*9880d681SAndroid Build Coastguard Worker  %shr = lshr i128 %r, %s
64*9880d681SAndroid Build Coastguard Worker  ret i128 %shr
65*9880d681SAndroid Build Coastguard Worker}
66