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 3 4 @A = global i32 zeroinitializer 5 @B = global i32 zeroinitializer 6 @C = global i32 zeroinitializer 7 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 14 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 32 } 33 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 43 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 68 } 69 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]] 77 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 90 } 91
