test/CodeGen/WebAssembly/select.ll

   1 ; RUN: llc < %s -asm-verbose=false -disable-wasm-fallthrough-return-opt -wasm-disable-explicit-locals -wasm-keep-registers | FileCheck %s --check-prefixes CHECK,SLOW
   2 ; RUN: llc < %s -asm-verbose=false -disable-wasm-fallthrough-return-opt -wasm-disable-explicit-locals -wasm-keep-registers -fast-isel -fast-isel-abort=1 | FileCheck %s
   3
   4 ; Test that wasm select instruction is selected from LLVM select instruction.
   5
   6 target datalayout = "e-m:e-p:32:32-i64:64-n32:64-S128"
   7 target triple = "wasm32-unknown-unknown"
   8
   9 ; CHECK-LABEL: select_i32_bool:
  10 ; CHECK-NEXT: .functype select_i32_bool (i32, i32, i32) -> (i32){{$}}
  11 ; CHECK-NEXT: i32.select $push0=, $1, $2, $0{{$}}
  12 ; CHECK-NEXT: return     $pop0{{$}}
  13 define i32 @select_i32_bool(i1 zeroext %a, i32 %b, i32 %c) {
  14   %cond = select i1 %a, i32 %b, i32 %c
  15   ret i32 %cond
  16 }
  17
  18 ; CHECK-LABEL: select_i32_bool_nozext:
  19 ; CHECK-NEXT: .functype select_i32_bool_nozext (i32, i32, i32) -> (i32){{$}}
  20 ; SLOW-NEXT: i32.select $push0=, $1, $2, $0{{$}}
  21 ; SLOW-NEXT: return     $pop0{{$}}
  22 define i32 @select_i32_bool_nozext(i1 %a, i32 %b, i32 %c) {
  23   %cond = select i1 %a, i32 %b, i32 %c
  24   ret i32 %cond
  25 }
  26
  27 ; CHECK-LABEL: select_i32_eq:
  28 ; CHECK-NEXT: .functype select_i32_eq (i32, i32, i32) -> (i32){{$}}
  29 ; CHECK-NEXT: i32.select $push0=, $2, $1, $0{{$}}
  30 ; CHECK-NEXT: return     $pop0{{$}}
  31 define i32 @select_i32_eq(i32 %a, i32 %b, i32 %c) {
  32   %cmp = icmp eq i32 %a, 0
  33   %cond = select i1 %cmp, i32 %b, i32 %c
  34   ret i32 %cond
  35 }
  36
  37 ; CHECK-LABEL: select_i32_ne:
  38 ; CHECK-NEXT: .functype select_i32_ne (i32, i32, i32) -> (i32){{$}}
  39 ; CHECK-NEXT: i32.select $push0=, $1, $2, $0{{$}}
  40 ; CHECK-NEXT: return     $pop0{{$}}
  41 define i32 @select_i32_ne(i32 %a, i32 %b, i32 %c) {
  42   %cmp = icmp ne i32 %a, 0
  43   %cond = select i1 %cmp, i32 %b, i32 %c
  44   ret i32 %cond
  45 }
  46
  47 ; CHECK-LABEL: select_i64_bool:
  48 ; CHECK-NEXT: .functype select_i64_bool (i32, i64, i64) -> (i64){{$}}
  49 ; CHECK-NEXT: i64.select $push0=, $1, $2, $0{{$}}
  50 ; CHECK-NEXT: return     $pop0{{$}}
  51 define i64 @select_i64_bool(i1 zeroext %a, i64 %b, i64 %c) {
  52   %cond = select i1 %a, i64 %b, i64 %c
  53   ret i64 %cond
  54 }
  55
  56 ; CHECK-LABEL: select_i64_bool_nozext:
  57 ; CHECK-NEXT: .functype select_i64_bool_nozext (i32, i64, i64) -> (i64){{$}}
  58 ; SLOW-NEXT: i64.select $push0=, $1, $2, $0{{$}}
  59 ; SLOW-NEXT: return     $pop0{{$}}
  60 define i64 @select_i64_bool_nozext(i1 %a, i64 %b, i64 %c) {
  61   %cond = select i1 %a, i64 %b, i64 %c
  62   ret i64 %cond
  63 }
  64
  65 ; CHECK-LABEL: select_i64_eq:
  66 ; CHECK-NEXT: .functype select_i64_eq (i32, i64, i64) -> (i64){{$}}
  67 ; CHECK-NEXT: i64.select $push0=, $2, $1, $0{{$}}
  68 ; CHECK-NEXT: return     $pop0{{$}}
  69 define i64 @select_i64_eq(i32 %a, i64 %b, i64 %c) {
  70   %cmp = icmp eq i32 %a, 0
  71   %cond = select i1 %cmp, i64 %b, i64 %c
  72   ret i64 %cond
  73 }
  74
  75 ; CHECK-LABEL: select_i64_ne:
  76 ; CHECK-NEXT: .functype select_i64_ne (i32, i64, i64) -> (i64){{$}}
  77 ; CHECK-NEXT: i64.select $push0=, $1, $2, $0{{$}}
  78 ; CHECK-NEXT: return     $pop0{{$}}
  79 define i64 @select_i64_ne(i32 %a, i64 %b, i64 %c) {
  80   %cmp = icmp ne i32 %a, 0
  81   %cond = select i1 %cmp, i64 %b, i64 %c
  82   ret i64 %cond
  83 }
  84
  85 ; CHECK-LABEL: select_f32_bool:
  86 ; CHECK-NEXT: .functype select_f32_bool (i32, f32, f32) -> (f32){{$}}
  87 ; CHECK-NEXT: f32.select $push0=, $1, $2, $0{{$}}
  88 ; CHECK-NEXT: return     $pop0{{$}}
  89 define float @select_f32_bool(i1 zeroext %a, float %b, float %c) {
  90   %cond = select i1 %a, float %b, float %c
  91   ret float %cond
  92 }
  93
  94 ; CHECK-LABEL: select_f32_bool_nozext:
  95 ; CHECK-NEXT: .functype select_f32_bool_nozext (i32, f32, f32) -> (f32){{$}}
  96 ; SLOW-NEXT: f32.select $push0=, $1, $2, $0{{$}}
  97 ; SLOW-NEXT: return     $pop0{{$}}
  98 define float @select_f32_bool_nozext(i1 %a, float %b, float %c) {
  99   %cond = select i1 %a, float %b, float %c
 100   ret float %cond
 101 }
 102
 103 ; CHECK-LABEL: select_f32_eq:
 104 ; CHECK-NEXT: .functype select_f32_eq (i32, f32, f32) -> (f32){{$}}
 105 ; CHECK-NEXT: f32.select $push0=, $2, $1, $0{{$}}
 106 ; CHECK-NEXT: return     $pop0{{$}}
 107 define float @select_f32_eq(i32 %a, float %b, float %c) {
 108   %cmp = icmp eq i32 %a, 0
 109   %cond = select i1 %cmp, float %b, float %c
 110   ret float %cond
 111 }
 112
 113 ; CHECK-LABEL: select_f32_ne:
 114 ; CHECK-NEXT: .functype select_f32_ne (i32, f32, f32) -> (f32){{$}}
 115 ; CHECK-NEXT: f32.select $push0=, $1, $2, $0{{$}}
 116 ; CHECK-NEXT: return     $pop0{{$}}
 117 define float @select_f32_ne(i32 %a, float %b, float %c) {
 118   %cmp = icmp ne i32 %a, 0
 119   %cond = select i1 %cmp, float %b, float %c
 120   ret float %cond
 121 }
 122
 123 ; CHECK-LABEL: select_f64_bool:
 124 ; CHECK-NEXT: .functype select_f64_bool (i32, f64, f64) -> (f64){{$}}
 125 ; CHECK-NEXT: f64.select $push0=, $1, $2, $0{{$}}
 126 ; CHECK-NEXT: return     $pop0{{$}}
 127 define double @select_f64_bool(i1 zeroext %a, double %b, double %c) {
 128   %cond = select i1 %a, double %b, double %c
 129   ret double %cond
 130 }
 131
 132 ; CHECK-LABEL: select_f64_bool_nozext:
 133 ; CHECK-NEXT: .functype select_f64_bool_nozext (i32, f64, f64) -> (f64){{$}}
 134 ; SLOW-NEXT: f64.select $push0=, $1, $2, $0{{$}}
 135 ; SLOW-NEXT: return     $pop0{{$}}
 136 define double @select_f64_bool_nozext(i1 %a, double %b, double %c) {
 137   %cond = select i1 %a, double %b, double %c
 138   ret double %cond
 139 }
 140
 141 ; CHECK-LABEL: select_f64_eq:
 142 ; CHECK-NEXT: .functype select_f64_eq (i32, f64, f64) -> (f64){{$}}
 143 ; CHECK-NEXT: f64.select $push0=, $2, $1, $0{{$}}
 144 ; CHECK-NEXT: return     $pop0{{$}}
 145 define double @select_f64_eq(i32 %a, double %b, double %c) {
 146   %cmp = icmp eq i32 %a, 0
 147   %cond = select i1 %cmp, double %b, double %c
 148   ret double %cond
 149 }
 150
 151 ; CHECK-LABEL: select_f64_ne:
 152 ; CHECK-NEXT: .functype select_f64_ne (i32, f64, f64) -> (f64){{$}}
 153 ; CHECK-NEXT: f64.select $push0=, $1, $2, $0{{$}}
 154 ; CHECK-NEXT: return     $pop0{{$}}
 155 define double @select_f64_ne(i32 %a, double %b, double %c) {
 156   %cmp = icmp ne i32 %a, 0
 157   %cond = select i1 %cmp, double %b, double %c
 158   ret double %cond
 159 }