[InstCombine] Signed saturation patterns
[llvm-complete.git] / test / CodeGen / AArch64 / and-sink.ll
blob91b7bd0db1726c541dcd24ff90b258362a1d75b0
1 ; RUN: llc -mtriple=aarch64-linux-gnu -verify-machineinstrs < %s | FileCheck %s
2 ; RUN: opt -S -codegenprepare -mtriple=aarch64-linux %s | FileCheck --check-prefix=CHECK-CGP %s
4 @A = global i32 zeroinitializer
5 @B = global i32 zeroinitializer
6 @C = global i32 zeroinitializer
8 ; Test that and is sunk into cmp block to form tbz.
9 define i32 @and_sink1(i32 %a, i1 %c) {
10 ; CHECK-LABEL: and_sink1:
11 ; CHECK: tbz w1, #0
12 ; CHECK: str wzr, [x{{[0-9]+}}, :lo12:A]
13 ; CHECK: tbnz {{w[0-9]+}}, #2
15 ; CHECK-CGP-LABEL: @and_sink1(
16 ; CHECK-CGP-NOT: and i32
17   %and = and i32 %a, 4
18   br i1 %c, label %bb0, label %bb2
19 bb0:
20 ; CHECK-CGP-LABEL: bb0:
21 ; CHECK-CGP: and i32
22 ; CHECK-CGP-NEXT: icmp eq i32
23 ; CHECK-CGP-NEXT: store
24 ; CHECK-CGP-NEXT: br
25   %cmp = icmp eq i32 %and, 0
26   store i32 0, i32* @A
27   br i1 %cmp, label %bb1, label %bb2
28 bb1:
29   ret i32 1
30 bb2:
31   ret i32 0
34 ; Test that both 'and' and cmp get sunk to form tbz.
35 define i32 @and_sink2(i32 %a, i1 %c, i1 %c2) {
36 ; CHECK-LABEL: and_sink2:
37 ; CHECK: str wzr, [x{{[0-9]+}}, :lo12:A]
38 ; CHECK: tbz w1, #0
39 ; CHECK: str wzr, [x{{[0-9]+}}, :lo12:B]
40 ; CHECK: tbz w2, #0
41 ; CHECK: str wzr, [x{{[0-9]+}}, :lo12:C]
42 ; CHECK: tbnz {{w[0-9]+}}, #2
44 ; CHECK-CGP-LABEL: @and_sink2(
45 ; CHECK-CGP-NOT: and i32
46   %and = and i32 %a, 4
47   store i32 0, i32* @A
48   br i1 %c, label %bb0, label %bb3
49 bb0:
50 ; CHECK-CGP-LABEL: bb0:
51 ; CHECK-CGP-NOT: and i32
52 ; CHECK-CGP-NOT: icmp
53   %cmp = icmp eq i32 %and, 0
54   store i32 0, i32* @B
55   br i1 %c2, label %bb1, label %bb3
56 bb1:
57 ; CHECK-CGP-LABEL: bb1:
58 ; CHECK-CGP: and i32
59 ; CHECK-CGP-NEXT: icmp eq i32
60 ; CHECK-CGP-NEXT: store
61 ; CHECK-CGP-NEXT: br
62   store i32 0, i32* @C
63   br i1 %cmp, label %bb2, label %bb0
64 bb2:
65   ret i32 1
66 bb3:
67   ret i32 0
70 ; Test that 'and' is not sunk since cbz is a better alternative.
71 define i32 @and_sink3(i32 %a) {
72 ; CHECK-LABEL: and_sink3:
73 ; CHECK: and [[REG:w[0-9]+]], w0, #0x3
74 ; CHECK: [[LOOP:.L[A-Z0-9_]+]]:
75 ; CHECK: str wzr, [x{{[0-9]+}}, :lo12:A]
76 ; CHECK: cbz [[REG]], [[LOOP]]
78 ; CHECK-CGP-LABEL: @and_sink3(
79 ; CHECK-CGP-NEXT: and i32
80   %and = and i32 %a, 3
81   br label %bb0
82 bb0:
83 ; CHECK-CGP-LABEL: bb0:
84 ; CHECK-CGP-NOT: and i32
85   %cmp = icmp eq i32 %and, 0
86   store i32 0, i32* @A
87   br i1 %cmp, label %bb0, label %bb2
88 bb2:
89   ret i32 0