llvm/test/CodeGen/AArch64/and-sink.ll

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