test/CodeGen/ARM/GlobalISel/arm-legalize-bitcounts.mir

   1 # RUN: llc -O0 -mtriple arm-linux-gnueabi -mattr=+v5t -run-pass=legalizer %s -o - | FileCheck %s -check-prefixes=CHECK,CLZ
   2 # RUN: llc -O0 -mtriple thumb-linux-gnueabi -mattr=+v6t2 -run-pass=legalizer %s -o - | FileCheck %s -check-prefixes=CHECK,CLZ
   3 # RUN: llc -O0 -mtriple arm-linux-gnueabi -mattr=-v5t -run-pass=legalizer %s -o - | FileCheck %s -check-prefixes=CHECK,LIBCALLS
   4 --- |
   5   define void @test_ctlz_s32() { ret void }
   6   define void @test_ctlz_zero_undef_s32() { ret void }
   7
   8   ; same as above but with extensions
   9   define void @test_ctlz_s16() { ret void }
  10   define void @test_ctlz_zero_undef_s8() { ret void }
  11 ...
  12 ---
  13 name:            test_ctlz_s32
  14 # CHECK-LABEL: name: test_ctlz_s32
  15 legalized:       false
  16 # CHECK: legalized: true
  17 regBankSelected: false
  18 selected:        false
  19 tracksRegLiveness: true
  20 registers:
  21   - { id: 0, class: _ }
  22   - { id: 1, class: _ }
  23 body:             |
  24   bb.0:
  25     liveins: $r0
  26
  27     ; CHECK: [[X:%[0-9]+]]:_(s32) = COPY $r0
  28     %0(s32) = COPY $r0
  29
  30     ; CLZ: [[R:%[0-9]+]]:_(s32) = G_CTLZ [[X]]
  31     ; LIBCALLS-NOT: G_CTLZ
  32     ; LIBCALLS: ADJCALLSTACKDOWN
  33     ; LIBCALLS: $r0 = COPY [[X]]
  34     ; LIBCALLS: BL &__clzsi2, {{.*}}, implicit $r0, implicit-def $r0
  35     ; LIBCALLS: [[COUNT:%[0-9]+]]:_(s32) = COPY $r0
  36     ; LIBCALLS: ADJCALLSTACKUP
  37     ; LIBCALLS-NOT: G_CTLZ
  38     ; LIBCALLS: [[ZERO:%[0-9]+]]:_(s32) = G_CONSTANT i32 0
  39     ; LIBCALLS: [[BITS:%[0-9]+]]:_(s32) = G_CONSTANT i32 32
  40     ; LIBCALLS: [[CMP:%[0-9]+]]:_(s1) = G_ICMP intpred(eq), [[X]](s32), [[ZERO]]
  41     ; LIBCALLS: [[R:%[0-9]+]]:_(s32) = G_SELECT [[CMP]](s1), [[BITS]], [[COUNT]]
  42     ; LIBCALLS-NOT: G_CTLZ
  43     %1(s32) = G_CTLZ %0
  44
  45     ; CHECK: $r0 = COPY [[R]]
  46     $r0 = COPY %1(s32)
  47     BX_RET 14, $noreg, implicit $r0
  48 ...
  49 ---
  50 name:            test_ctlz_zero_undef_s32
  51 # CHECK-LABEL: name: test_ctlz_zero_undef_s32
  52 legalized:       false
  53 # CHECK: legalized: true
  54 regBankSelected: false
  55 selected:        false
  56 tracksRegLiveness: true
  57 registers:
  58   - { id: 0, class: _ }
  59   - { id: 1, class: _ }
  60 body:             |
  61   bb.0:
  62     liveins: $r0
  63
  64     ; CHECK: [[X:%[0-9]+]]:_(s32) = COPY $r0
  65     %0(s32) = COPY $r0
  66
  67     ; CLZ: [[R:%[0-9]+]]:_(s32) = G_CTLZ [[X]]
  68     ; LIBCALLS-NOT: G_CTLZ
  69     ; LIBCALLS: ADJCALLSTACKDOWN
  70     ; LIBCALLS: $r0 = COPY [[X]]
  71     ; LIBCALLS: BL &__clzsi2, {{.*}}, implicit $r0, implicit-def $r0
  72     ; LIBCALLS: [[R:%[0-9]+]]:_(s32) = COPY $r0
  73     ; LIBCALLS: ADJCALLSTACKUP
  74     ; LIBCALLS-NOT: G_CTLZ
  75     %1(s32) = G_CTLZ_ZERO_UNDEF %0
  76
  77     ; CHECK: $r0 = COPY [[R]]
  78     $r0 = COPY %1(s32)
  79     BX_RET 14, $noreg, implicit $r0
  80 ...
  81 ---
  82 name:            test_ctlz_s16
  83 # CHECK-LABEL: name: test_ctlz_s16
  84 legalized:       false
  85 # CHECK: legalized: true
  86 regBankSelected: false
  87 selected:        false
  88 tracksRegLiveness: true
  89 registers:
  90   - { id: 0, class: _ }
  91   - { id: 1, class: _ }
  92   - { id: 2, class: _ }
  93   - { id: 3, class: _ }
  94 body:             |
  95   bb.0:
  96     liveins: $r0
  97
  98     ; CHECK: [[X:%[0-9]+]]:_(s32) = COPY $r0
  99     ; CHECK: [[BITMASK:%[0-9]+]]:_(s32) = G_CONSTANT i32 65535
 100     ; CHECK: [[XAGAIN:%[0-9]+]]:_(s32) = COPY [[X]]
 101     ; CHECK: [[X32:%[0-9]+]]:_(s32) = G_AND [[XAGAIN]], [[BITMASK]]
 102     %0(s32) = COPY $r0
 103     %1(s16) = G_TRUNC %0(s32)
 104
 105     ; Check that the operation is performed for 32 bits
 106     ; CLZ: [[COUNT:%[0-9]+]]:_(s32) = G_CTLZ [[X32]]
 107     ; LIBCALLS-NOT: G_CTLZ
 108     ; LIBCALLS: ADJCALLSTACKDOWN
 109     ; LIBCALLS: $r0 = COPY [[X32]]
 110     ; LIBCALLS: BL &__clzsi2, {{.*}}, implicit $r0, implicit-def $r0
 111     ; LIBCALLS: [[UNDEFCOUNT:%[0-9]+]]:_(s32) = COPY $r0
 112     ; LIBCALLS: ADJCALLSTACKUP
 113     ; LIBCALLS-NOT: G_CTLZ
 114     ; LIBCALLS: [[ZERO:%[0-9]+]]:_(s32) = G_CONSTANT i32 0
 115     ; LIBCALLS: [[BITS:%[0-9]+]]:_(s32) = G_CONSTANT i32 32
 116     ; LIBCALLS: [[CMP:%[0-9]+]]:_(s1) = G_ICMP intpred(eq), {{%[0-9]+}}(s32), [[ZERO]]
 117     ; LIBCALLS: [[COUNT:%[0-9]+]]:_(s32) = G_SELECT [[CMP]](s1), [[BITS]], [[UNDEFCOUNT]]
 118     ; LIBCALLS-NOT: G_CTLZ
 119     ; CHECK: [[BITDIFF:%[0-9]+]]:_(s32) = G_CONSTANT i32 16
 120     ; CHECK: [[R32:%[0-9]+]]:_(s32) = G_SUB [[COUNT]], [[BITDIFF]]
 121     %2(s16) = G_CTLZ %1
 122
 123     ; CHECK: [[RAGAIN:%[0-9]+]]:_(s32) = COPY [[R32]]
 124     ; CHECK: [[SHIFTEDR:%[0-9]+]]:_(s32) = G_SHL [[RAGAIN]], [[BITDIFF]]
 125     ; CHECK: [[R:%[0-9]+]]:_(s32) = G_ASHR [[SHIFTEDR]], [[BITDIFF]]
 126     ; CHECK: $r0 = COPY [[R]]
 127     %3(s32) = G_SEXT %2(s16)
 128     $r0 = COPY %3(s32)
 129     BX_RET 14, $noreg, implicit $r0
 130 ...
 131 ---
 132 name:            test_ctlz_zero_undef_s8
 133 # CHECK-LABEL: name: test_ctlz_zero_undef_s8
 134 legalized:       false
 135 # CHECK: legalized: true
 136 regBankSelected: false
 137 selected:        false
 138 tracksRegLiveness: true
 139 registers:
 140   - { id: 0, class: _ }
 141   - { id: 1, class: _ }
 142   - { id: 2, class: _ }
 143   - { id: 3, class: _ }
 144 body:             |
 145   bb.0:
 146     liveins: $r0
 147
 148     ; CHECK: [[X:%[0-9]+]]:_(s32) = COPY $r0
 149     ; CHECK: [[BITMASK:%[0-9]+]]:_(s32) = G_CONSTANT i32 255
 150     ; CHECK: [[XAGAIN:%[0-9]+]]:_(s32) = COPY [[X]]
 151     ; CHECK: [[X32:%[0-9]+]]:_(s32) = G_AND [[XAGAIN]], [[BITMASK]]
 152     %0(s32) = COPY $r0
 153     %1(s8) = G_TRUNC %0(s32)
 154
 155     ; Check that the operation is performed for 32 bits
 156     ; CLZ: [[COUNT:%[0-9]+]]:_(s32) = G_CTLZ
 157     ; CLZ-NOT: G_CTLZ_ZERO_UNDEF
 158     ; LIBCALLS-NOT: G_CTLZ
 159     ; LIBCALLS: ADJCALLSTACKDOWN
 160     ; LIBCALLS: $r0 = COPY [[X32]]
 161     ; LIBCALLS: BL &__clzsi2, {{.*}}, implicit $r0, implicit-def $r0
 162     ; LIBCALLS: [[COUNT:%[0-9]+]]:_(s32) = COPY $r0
 163     ; LIBCALLS: ADJCALLSTACKUP
 164     ; LIBCALLS-NOT: G_CTLZ
 165     ; CHECK: [[BITDIFF:%[0-9]+]]:_(s32) = G_CONSTANT i32 24
 166     ; CHECK: [[R32:%[0-9]+]]:_(s32) = G_SUB [[COUNT]], [[BITDIFF]]
 167     %2(s8) = G_CTLZ_ZERO_UNDEF %1
 168
 169     ; CHECK: [[RAGAIN:%[0-9]+]]:_(s32) = COPY [[R32]]
 170     ; CHECK: [[SHIFTEDR:%[0-9]+]]:_(s32) = G_SHL [[RAGAIN]], [[BITDIFF]]
 171     ; CHECK: [[R:%[0-9]+]]:_(s32) = G_ASHR [[SHIFTEDR]], [[BITDIFF]]
 172     ; CHECK: $r0 = COPY [[R]]
 173     %3(s32) = G_SEXT %2(s8)
 174     $r0 = COPY %3(s32)
 175     BX_RET 14, $noreg, implicit $r0
 176 ...