llvm/test/CodeGen/WebAssembly/select.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc < %s -disable-wasm-fallthrough-return-opt -wasm-disable-explicit-locals -wasm-keep-registers | FileCheck %s
   3 ; RUN: llc < %s -disable-wasm-fallthrough-return-opt -wasm-disable-explicit-locals -wasm-keep-registers -fast-isel -fast-isel-abort=1 | FileCheck %s --check-prefix=FAST
   4
   5 ; Test that wasm select instruction is selected from LLVM select instruction.
   6
   7 target triple = "wasm32-unknown-unknown"
   8
   9 define i32 @select_i32_bool(i1 zeroext %a, i32 %b, i32 %c) {
  10 ; CHECK-LABEL: select_i32_bool:
  11 ; CHECK:         .functype select_i32_bool (i32, i32, i32) -> (i32)
  12 ; CHECK-NEXT:  # %bb.0:
  13 ; CHECK-NEXT:    i32.select $push0=, $1, $2, $0
  14 ; CHECK-NEXT:    return $pop0
  15 ;
  16 ; FAST-LABEL: select_i32_bool:
  17 ; FAST:         .functype select_i32_bool (i32, i32, i32) -> (i32)
  18 ; FAST-NEXT:  # %bb.0:
  19 ; FAST-NEXT:    i32.select $push0=, $1, $2, $0
  20 ; FAST-NEXT:    return $pop0
  21   %cond = select i1 %a, i32 %b, i32 %c
  22   ret i32 %cond
  23 }
  24
  25 define i32 @select_i32_bool_nozext(i1 %a, i32 %b, i32 %c) {
  26 ; CHECK-LABEL: select_i32_bool_nozext:
  27 ; CHECK:         .functype select_i32_bool_nozext (i32, i32, i32) -> (i32)
  28 ; CHECK-NEXT:  # %bb.0:
  29 ; CHECK-NEXT:    i32.const $push0=, 1
  30 ; CHECK-NEXT:    i32.and $push1=, $0, $pop0
  31 ; CHECK-NEXT:    i32.select $push2=, $1, $2, $pop1
  32 ; CHECK-NEXT:    return $pop2
  33 ;
  34 ; FAST-LABEL: select_i32_bool_nozext:
  35 ; FAST:         .functype select_i32_bool_nozext (i32, i32, i32) -> (i32)
  36 ; FAST-NEXT:  # %bb.0:
  37 ; FAST-NEXT:    i32.const $push0=, 1
  38 ; FAST-NEXT:    i32.and $push1=, $0, $pop0
  39 ; FAST-NEXT:    i32.select $push2=, $1, $2, $pop1
  40 ; FAST-NEXT:    return $pop2
  41   %cond = select i1 %a, i32 %b, i32 %c
  42   ret i32 %cond
  43 }
  44
  45 define i32 @select_i32_eq(i32 %a, i32 %b, i32 %c) {
  46 ; CHECK-LABEL: select_i32_eq:
  47 ; CHECK:         .functype select_i32_eq (i32, i32, i32) -> (i32)
  48 ; CHECK-NEXT:  # %bb.0:
  49 ; CHECK-NEXT:    i32.select $push0=, $2, $1, $0
  50 ; CHECK-NEXT:    return $pop0
  51 ;
  52 ; FAST-LABEL: select_i32_eq:
  53 ; FAST:         .functype select_i32_eq (i32, i32, i32) -> (i32)
  54 ; FAST-NEXT:  # %bb.0:
  55 ; FAST-NEXT:    i32.select $push0=, $2, $1, $0
  56 ; FAST-NEXT:    return $pop0
  57   %cmp = icmp eq i32 %a, 0
  58   %cond = select i1 %cmp, i32 %b, i32 %c
  59   ret i32 %cond
  60 }
  61
  62 define i32 @select_i32_ne(i32 %a, i32 %b, i32 %c) {
  63 ; CHECK-LABEL: select_i32_ne:
  64 ; CHECK:         .functype select_i32_ne (i32, i32, i32) -> (i32)
  65 ; CHECK-NEXT:  # %bb.0:
  66 ; CHECK-NEXT:    i32.select $push0=, $1, $2, $0
  67 ; CHECK-NEXT:    return $pop0
  68 ;
  69 ; FAST-LABEL: select_i32_ne:
  70 ; FAST:         .functype select_i32_ne (i32, i32, i32) -> (i32)
  71 ; FAST-NEXT:  # %bb.0:
  72 ; FAST-NEXT:    i32.select $push0=, $1, $2, $0
  73 ; FAST-NEXT:    return $pop0
  74   %cmp = icmp ne i32 %a, 0
  75   %cond = select i1 %cmp, i32 %b, i32 %c
  76   ret i32 %cond
  77 }
  78
  79 define i64 @select_i64_bool(i1 zeroext %a, i64 %b, i64 %c) {
  80 ; CHECK-LABEL: select_i64_bool:
  81 ; CHECK:         .functype select_i64_bool (i32, i64, i64) -> (i64)
  82 ; CHECK-NEXT:  # %bb.0:
  83 ; CHECK-NEXT:    i64.select $push0=, $1, $2, $0
  84 ; CHECK-NEXT:    return $pop0
  85 ;
  86 ; FAST-LABEL: select_i64_bool:
  87 ; FAST:         .functype select_i64_bool (i32, i64, i64) -> (i64)
  88 ; FAST-NEXT:  # %bb.0:
  89 ; FAST-NEXT:    i64.select $push0=, $1, $2, $0
  90 ; FAST-NEXT:    return $pop0
  91   %cond = select i1 %a, i64 %b, i64 %c
  92   ret i64 %cond
  93 }
  94
  95 define i64 @select_i64_bool_nozext(i1 %a, i64 %b, i64 %c) {
  96 ; CHECK-LABEL: select_i64_bool_nozext:
  97 ; CHECK:         .functype select_i64_bool_nozext (i32, i64, i64) -> (i64)
  98 ; CHECK-NEXT:  # %bb.0:
  99 ; CHECK-NEXT:    i32.const $push0=, 1
 100 ; CHECK-NEXT:    i32.and $push1=, $0, $pop0
 101 ; CHECK-NEXT:    i64.select $push2=, $1, $2, $pop1
 102 ; CHECK-NEXT:    return $pop2
 103 ;
 104 ; FAST-LABEL: select_i64_bool_nozext:
 105 ; FAST:         .functype select_i64_bool_nozext (i32, i64, i64) -> (i64)
 106 ; FAST-NEXT:  # %bb.0:
 107 ; FAST-NEXT:    i32.const $push0=, 1
 108 ; FAST-NEXT:    i32.and $push1=, $0, $pop0
 109 ; FAST-NEXT:    i64.select $push2=, $1, $2, $pop1
 110 ; FAST-NEXT:    return $pop2
 111   %cond = select i1 %a, i64 %b, i64 %c
 112   ret i64 %cond
 113 }
 114
 115 define i64 @select_i64_eq(i32 %a, i64 %b, i64 %c) {
 116 ; CHECK-LABEL: select_i64_eq:
 117 ; CHECK:         .functype select_i64_eq (i32, i64, i64) -> (i64)
 118 ; CHECK-NEXT:  # %bb.0:
 119 ; CHECK-NEXT:    i64.select $push0=, $2, $1, $0
 120 ; CHECK-NEXT:    return $pop0
 121 ;
 122 ; FAST-LABEL: select_i64_eq:
 123 ; FAST:         .functype select_i64_eq (i32, i64, i64) -> (i64)
 124 ; FAST-NEXT:  # %bb.0:
 125 ; FAST-NEXT:    i64.select $push0=, $2, $1, $0
 126 ; FAST-NEXT:    return $pop0
 127   %cmp = icmp eq i32 %a, 0
 128   %cond = select i1 %cmp, i64 %b, i64 %c
 129   ret i64 %cond
 130 }
 131
 132 define i64 @select_i64_ne(i32 %a, i64 %b, i64 %c) {
 133 ; CHECK-LABEL: select_i64_ne:
 134 ; CHECK:         .functype select_i64_ne (i32, i64, i64) -> (i64)
 135 ; CHECK-NEXT:  # %bb.0:
 136 ; CHECK-NEXT:    i64.select $push0=, $1, $2, $0
 137 ; CHECK-NEXT:    return $pop0
 138 ;
 139 ; FAST-LABEL: select_i64_ne:
 140 ; FAST:         .functype select_i64_ne (i32, i64, i64) -> (i64)
 141 ; FAST-NEXT:  # %bb.0:
 142 ; FAST-NEXT:    i64.select $push0=, $1, $2, $0
 143 ; FAST-NEXT:    return $pop0
 144   %cmp = icmp ne i32 %a, 0
 145   %cond = select i1 %cmp, i64 %b, i64 %c
 146   ret i64 %cond
 147 }
 148
 149 define float @select_f32_bool(i1 zeroext %a, float %b, float %c) {
 150 ; CHECK-LABEL: select_f32_bool:
 151 ; CHECK:         .functype select_f32_bool (i32, f32, f32) -> (f32)
 152 ; CHECK-NEXT:  # %bb.0:
 153 ; CHECK-NEXT:    f32.select $push0=, $1, $2, $0
 154 ; CHECK-NEXT:    return $pop0
 155 ;
 156 ; FAST-LABEL: select_f32_bool:
 157 ; FAST:         .functype select_f32_bool (i32, f32, f32) -> (f32)
 158 ; FAST-NEXT:  # %bb.0:
 159 ; FAST-NEXT:    f32.select $push0=, $1, $2, $0
 160 ; FAST-NEXT:    return $pop0
 161   %cond = select i1 %a, float %b, float %c
 162   ret float %cond
 163 }
 164
 165 define float @select_f32_bool_nozext(i1 %a, float %b, float %c) {
 166 ; CHECK-LABEL: select_f32_bool_nozext:
 167 ; CHECK:         .functype select_f32_bool_nozext (i32, f32, f32) -> (f32)
 168 ; CHECK-NEXT:  # %bb.0:
 169 ; CHECK-NEXT:    i32.const $push0=, 1
 170 ; CHECK-NEXT:    i32.and $push1=, $0, $pop0
 171 ; CHECK-NEXT:    f32.select $push2=, $1, $2, $pop1
 172 ; CHECK-NEXT:    return $pop2
 173 ;
 174 ; FAST-LABEL: select_f32_bool_nozext:
 175 ; FAST:         .functype select_f32_bool_nozext (i32, f32, f32) -> (f32)
 176 ; FAST-NEXT:  # %bb.0:
 177 ; FAST-NEXT:    i32.const $push0=, 1
 178 ; FAST-NEXT:    i32.and $push1=, $0, $pop0
 179 ; FAST-NEXT:    f32.select $push2=, $1, $2, $pop1
 180 ; FAST-NEXT:    return $pop2
 181   %cond = select i1 %a, float %b, float %c
 182   ret float %cond
 183 }
 184
 185 define float @select_f32_eq(i32 %a, float %b, float %c) {
 186 ; CHECK-LABEL: select_f32_eq:
 187 ; CHECK:         .functype select_f32_eq (i32, f32, f32) -> (f32)
 188 ; CHECK-NEXT:  # %bb.0:
 189 ; CHECK-NEXT:    f32.select $push0=, $2, $1, $0
 190 ; CHECK-NEXT:    return $pop0
 191 ;
 192 ; FAST-LABEL: select_f32_eq:
 193 ; FAST:         .functype select_f32_eq (i32, f32, f32) -> (f32)
 194 ; FAST-NEXT:  # %bb.0:
 195 ; FAST-NEXT:    f32.select $push0=, $2, $1, $0
 196 ; FAST-NEXT:    return $pop0
 197   %cmp = icmp eq i32 %a, 0
 198   %cond = select i1 %cmp, float %b, float %c
 199   ret float %cond
 200 }
 201
 202 define float @select_f32_ne(i32 %a, float %b, float %c) {
 203 ; CHECK-LABEL: select_f32_ne:
 204 ; CHECK:         .functype select_f32_ne (i32, f32, f32) -> (f32)
 205 ; CHECK-NEXT:  # %bb.0:
 206 ; CHECK-NEXT:    f32.select $push0=, $1, $2, $0
 207 ; CHECK-NEXT:    return $pop0
 208 ;
 209 ; FAST-LABEL: select_f32_ne:
 210 ; FAST:         .functype select_f32_ne (i32, f32, f32) -> (f32)
 211 ; FAST-NEXT:  # %bb.0:
 212 ; FAST-NEXT:    f32.select $push0=, $1, $2, $0
 213 ; FAST-NEXT:    return $pop0
 214   %cmp = icmp ne i32 %a, 0
 215   %cond = select i1 %cmp, float %b, float %c
 216   ret float %cond
 217 }
 218
 219 define double @select_f64_bool(i1 zeroext %a, double %b, double %c) {
 220 ; CHECK-LABEL: select_f64_bool:
 221 ; CHECK:         .functype select_f64_bool (i32, f64, f64) -> (f64)
 222 ; CHECK-NEXT:  # %bb.0:
 223 ; CHECK-NEXT:    f64.select $push0=, $1, $2, $0
 224 ; CHECK-NEXT:    return $pop0
 225 ;
 226 ; FAST-LABEL: select_f64_bool:
 227 ; FAST:         .functype select_f64_bool (i32, f64, f64) -> (f64)
 228 ; FAST-NEXT:  # %bb.0:
 229 ; FAST-NEXT:    f64.select $push0=, $1, $2, $0
 230 ; FAST-NEXT:    return $pop0
 231   %cond = select i1 %a, double %b, double %c
 232   ret double %cond
 233 }
 234
 235 define double @select_f64_bool_nozext(i1 %a, double %b, double %c) {
 236 ; CHECK-LABEL: select_f64_bool_nozext:
 237 ; CHECK:         .functype select_f64_bool_nozext (i32, f64, f64) -> (f64)
 238 ; CHECK-NEXT:  # %bb.0:
 239 ; CHECK-NEXT:    i32.const $push0=, 1
 240 ; CHECK-NEXT:    i32.and $push1=, $0, $pop0
 241 ; CHECK-NEXT:    f64.select $push2=, $1, $2, $pop1
 242 ; CHECK-NEXT:    return $pop2
 243 ;
 244 ; FAST-LABEL: select_f64_bool_nozext:
 245 ; FAST:         .functype select_f64_bool_nozext (i32, f64, f64) -> (f64)
 246 ; FAST-NEXT:  # %bb.0:
 247 ; FAST-NEXT:    i32.const $push0=, 1
 248 ; FAST-NEXT:    i32.and $push1=, $0, $pop0
 249 ; FAST-NEXT:    f64.select $push2=, $1, $2, $pop1
 250 ; FAST-NEXT:    return $pop2
 251   %cond = select i1 %a, double %b, double %c
 252   ret double %cond
 253 }
 254
 255 define double @select_f64_eq(i32 %a, double %b, double %c) {
 256 ; CHECK-LABEL: select_f64_eq:
 257 ; CHECK:         .functype select_f64_eq (i32, f64, f64) -> (f64)
 258 ; CHECK-NEXT:  # %bb.0:
 259 ; CHECK-NEXT:    f64.select $push0=, $2, $1, $0
 260 ; CHECK-NEXT:    return $pop0
 261 ;
 262 ; FAST-LABEL: select_f64_eq:
 263 ; FAST:         .functype select_f64_eq (i32, f64, f64) -> (f64)
 264 ; FAST-NEXT:  # %bb.0:
 265 ; FAST-NEXT:    f64.select $push0=, $2, $1, $0
 266 ; FAST-NEXT:    return $pop0
 267   %cmp = icmp eq i32 %a, 0
 268   %cond = select i1 %cmp, double %b, double %c
 269   ret double %cond
 270 }
 271
 272 define double @select_f64_ne(i32 %a, double %b, double %c) {
 273 ; CHECK-LABEL: select_f64_ne:
 274 ; CHECK:         .functype select_f64_ne (i32, f64, f64) -> (f64)
 275 ; CHECK-NEXT:  # %bb.0:
 276 ; CHECK-NEXT:    f64.select $push0=, $1, $2, $0
 277 ; CHECK-NEXT:    return $pop0
 278 ;
 279 ; FAST-LABEL: select_f64_ne:
 280 ; FAST:         .functype select_f64_ne (i32, f64, f64) -> (f64)
 281 ; FAST-NEXT:  # %bb.0:
 282 ; FAST-NEXT:    f64.select $push0=, $1, $2, $0
 283 ; FAST-NEXT:    return $pop0
 284   %cmp = icmp ne i32 %a, 0
 285   %cond = select i1 %cmp, double %b, double %c
 286   ret double %cond
 287 }
 288
 289 define i32 @pr40805_i32(i32 %x, i32 %y, i32 %z) {
 290 ; CHECK-LABEL: pr40805_i32:
 291 ; CHECK:         .functype pr40805_i32 (i32, i32, i32) -> (i32)
 292 ; CHECK-NEXT:  # %bb.0:
 293 ; CHECK-NEXT:    i32.const $push0=, 1
 294 ; CHECK-NEXT:    i32.and $push1=, $0, $pop0
 295 ; CHECK-NEXT:    i32.select $push2=, $1, $2, $pop1
 296 ; CHECK-NEXT:    return $pop2
 297 ;
 298 ; FAST-LABEL: pr40805_i32:
 299 ; FAST:         .functype pr40805_i32 (i32, i32, i32) -> (i32)
 300 ; FAST-NEXT:  # %bb.0:
 301 ; FAST-NEXT:    i32.const $push1=, 1
 302 ; FAST-NEXT:    i32.and $push2=, $0, $pop1
 303 ; FAST-NEXT:    i32.select $push0=, $1, $2, $pop2
 304 ; FAST-NEXT:    return $pop0
 305   %a = and i32 %x, 1
 306   %b = icmp ne i32 %a, 0
 307   %c = select i1 %b, i32 %y, i32 %z
 308   ret i32 %c
 309 }
 310
 311 define i64 @pr40805_i64(i64 %x, i64 %y, i64 %z) {
 312 ; CHECK-LABEL: pr40805_i64:
 313 ; CHECK:         .functype pr40805_i64 (i64, i64, i64) -> (i64)
 314 ; CHECK-NEXT:  # %bb.0:
 315 ; CHECK-NEXT:    i32.wrap_i64 $push0=, $0
 316 ; CHECK-NEXT:    i32.const $push1=, 1
 317 ; CHECK-NEXT:    i32.and $push2=, $pop0, $pop1
 318 ; CHECK-NEXT:    i64.select $push3=, $1, $2, $pop2
 319 ; CHECK-NEXT:    return $pop3
 320 ;
 321 ; FAST-LABEL: pr40805_i64:
 322 ; FAST:         .functype pr40805_i64 (i64, i64, i64) -> (i64)
 323 ; FAST-NEXT:  # %bb.0:
 324 ; FAST-NEXT:    i64.const $push5=, 1
 325 ; FAST-NEXT:    i64.and $push6=, $0, $pop5
 326 ; FAST-NEXT:    i64.const $push3=, 0
 327 ; FAST-NEXT:    i64.ne $push4=, $pop6, $pop3
 328 ; FAST-NEXT:    i32.const $push0=, 1
 329 ; FAST-NEXT:    i32.and $push1=, $pop4, $pop0
 330 ; FAST-NEXT:    i64.select $push2=, $1, $2, $pop1
 331 ; FAST-NEXT:    return $pop2
 332   %a = and i64 %x, 1
 333   %b = icmp ne i64 %a, 0
 334   %c = select i1 %b, i64 %y, i64 %z
 335   ret i64 %c
 336 }
 337
 338 define float @pr44012_i32(i32 %x, float %y, float %z) {
 339 ; CHECK-LABEL: pr44012_i32:
 340 ; CHECK:         .functype pr44012_i32 (i32, f32, f32) -> (f32)
 341 ; CHECK-NEXT:  # %bb.0:
 342 ; CHECK-NEXT:    i32.const $push0=, 1
 343 ; CHECK-NEXT:    i32.and $push1=, $0, $pop0
 344 ; CHECK-NEXT:    f32.select $push2=, $1, $2, $pop1
 345 ; CHECK-NEXT:    return $pop2
 346 ;
 347 ; FAST-LABEL: pr44012_i32:
 348 ; FAST:         .functype pr44012_i32 (i32, f32, f32) -> (f32)
 349 ; FAST-NEXT:  # %bb.0:
 350 ; FAST-NEXT:    i32.const $push1=, 1
 351 ; FAST-NEXT:    i32.and $push2=, $0, $pop1
 352 ; FAST-NEXT:    f32.select $push0=, $1, $2, $pop2
 353 ; FAST-NEXT:    return $pop0
 354   %a = and i32 %x, 1
 355   %b = icmp ne i32 %a, 0
 356   %c = select i1 %b, float %y, float %z
 357   ret float %c
 358 }
 359
 360 define float @pr44012_i64(i64 %x, float %y, float %z) {
 361 ; CHECK-LABEL: pr44012_i64:
 362 ; CHECK:         .functype pr44012_i64 (i64, f32, f32) -> (f32)
 363 ; CHECK-NEXT:  # %bb.0:
 364 ; CHECK-NEXT:    i32.wrap_i64 $push0=, $0
 365 ; CHECK-NEXT:    i32.const $push1=, 1
 366 ; CHECK-NEXT:    i32.and $push2=, $pop0, $pop1
 367 ; CHECK-NEXT:    f32.select $push3=, $1, $2, $pop2
 368 ; CHECK-NEXT:    return $pop3
 369 ;
 370 ; FAST-LABEL: pr44012_i64:
 371 ; FAST:         .functype pr44012_i64 (i64, f32, f32) -> (f32)
 372 ; FAST-NEXT:  # %bb.0:
 373 ; FAST-NEXT:    i64.const $push5=, 1
 374 ; FAST-NEXT:    i64.and $push6=, $0, $pop5
 375 ; FAST-NEXT:    i64.const $push3=, 0
 376 ; FAST-NEXT:    i64.ne $push4=, $pop6, $pop3
 377 ; FAST-NEXT:    i32.const $push0=, 1
 378 ; FAST-NEXT:    i32.and $push1=, $pop4, $pop0
 379 ; FAST-NEXT:    f32.select $push2=, $1, $2, $pop1
 380 ; FAST-NEXT:    return $pop2
 381   %a = and i64 %x, 1
 382   %b = icmp ne i64 %a, 0
 383   %c = select i1 %b, float %y, float %z
 384   ret float %c
 385 }