llvm/test/CodeGen/VE/Vector/vp_select.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc < %s -march=ve -mattr=+vpu | FileCheck %s
   3
   4 declare <256 x i32> @llvm.vp.select.v256i32(<256 x i1>, <256 x i32>, <256 x i32>, i32)
   5
   6 define fastcc <256 x i32> @test_vp_select_v256i32_vv(<256 x i32> %i0, <256 x i32> %i1, <256 x i1> %m, i32 %pivot) {
   7 ; CHECK-LABEL: test_vp_select_v256i32_vv:
   8 ; CHECK:       # %bb.0:
   9 ; CHECK-NEXT:    and %s0, %s0, (32)0
  10 ; CHECK-NEXT:    lvl %s0
  11 ; CHECK-NEXT:    vmrg %v1, %v1, %v0, %vm1
  12 ; CHECK-NEXT:    lea %s16, 256
  13 ; CHECK-NEXT:    lvl %s16
  14 ; CHECK-NEXT:    vor %v0, (0)1, %v1
  15 ; CHECK-NEXT:    b.l.t (, %s10)
  16   %r0 = call <256 x i32> @llvm.vp.select.v256i32(<256 x i1> %m, <256 x i32> %i0, <256 x i32> %i1, i32 %pivot)
  17   ret <256 x i32> %r0
  18 }
  19
  20 define fastcc <256 x i32> @test_vp_select_v256i32_vr(<256 x i32> %i0, i32 %s1, <256 x i1> %m, i32 %pivot) {
  21 ; CHECK-LABEL: test_vp_select_v256i32_vr:
  22 ; CHECK:       # %bb.0:
  23 ; CHECK-NEXT:    and %s1, %s1, (32)0
  24 ; CHECK-NEXT:    and %s0, %s0, (32)0
  25 ; CHECK-NEXT:    lea %s2, 256
  26 ; CHECK-NEXT:    lvl %s2
  27 ; CHECK-NEXT:    vbrd %v1, %s0
  28 ; CHECK-NEXT:    lvl %s1
  29 ; CHECK-NEXT:    vmrg %v1, %v1, %v0, %vm1
  30 ; CHECK-NEXT:    lea %s16, 256
  31 ; CHECK-NEXT:    lvl %s16
  32 ; CHECK-NEXT:    vor %v0, (0)1, %v1
  33 ; CHECK-NEXT:    b.l.t (, %s10)
  34   %xins = insertelement <256 x i32> undef, i32 %s1, i32 0
  35   %i1 = shufflevector <256 x i32> %xins, <256 x i32> undef, <256 x i32> zeroinitializer
  36   %r0 = call <256 x i32> @llvm.vp.select.v256i32(<256 x i1> %m, <256 x i32> %i0, <256 x i32> %i1, i32 %pivot)
  37   ret <256 x i32> %r0
  38 }
  39
  40 declare <256 x float> @llvm.vp.select.v256f32(<256 x i1>, <256 x float>, <256 x float>, i32)
  41
  42 define fastcc <256 x float> @test_vp_select_v256f32_vv(<256 x float> %i0, <256 x float> %i1, <256 x i1> %m, i32 %pivot) {
  43 ; CHECK-LABEL: test_vp_select_v256f32_vv:
  44 ; CHECK:       # %bb.0:
  45 ; CHECK-NEXT:    and %s0, %s0, (32)0
  46 ; CHECK-NEXT:    lvl %s0
  47 ; CHECK-NEXT:    vmrg %v1, %v1, %v0, %vm1
  48 ; CHECK-NEXT:    lea %s16, 256
  49 ; CHECK-NEXT:    lvl %s16
  50 ; CHECK-NEXT:    vor %v0, (0)1, %v1
  51 ; CHECK-NEXT:    b.l.t (, %s10)
  52   %r0 = call <256 x float> @llvm.vp.select.v256f32(<256 x i1> %m, <256 x float> %i0, <256 x float> %i1, i32 %pivot)
  53   ret <256 x float> %r0
  54 }
  55
  56 define fastcc <256 x float> @test_vp_select_v256f32_vr(<256 x float> %i0, float %s1, <256 x i1> %m, i32 %pivot) {
  57 ; CHECK-LABEL: test_vp_select_v256f32_vr:
  58 ; CHECK:       # %bb.0:
  59 ; CHECK-NEXT:    and %s1, %s1, (32)0
  60 ; CHECK-NEXT:    lea %s2, 256
  61 ; CHECK-NEXT:    lvl %s2
  62 ; CHECK-NEXT:    vbrd %v1, %s0
  63 ; CHECK-NEXT:    lvl %s1
  64 ; CHECK-NEXT:    vmrg %v1, %v1, %v0, %vm1
  65 ; CHECK-NEXT:    lea %s16, 256
  66 ; CHECK-NEXT:    lvl %s16
  67 ; CHECK-NEXT:    vor %v0, (0)1, %v1
  68 ; CHECK-NEXT:    b.l.t (, %s10)
  69   %xins = insertelement <256 x float> undef, float %s1, i32 0
  70   %i1 = shufflevector <256 x float> %xins, <256 x float> undef, <256 x i32> zeroinitializer
  71   %r0 = call <256 x float> @llvm.vp.select.v256f32(<256 x i1> %m, <256 x float> %i0, <256 x float> %i1, i32 %pivot)
  72   ret <256 x float> %r0
  73 }
  74
  75 declare <256 x double> @llvm.vp.select.v256f64(<256 x i1>, <256 x double>, <256 x double>, i32)
  76
  77 define fastcc <256 x double> @test_vp_select_v256f64_vv(<256 x double> %i0, <256 x double> %i1, <256 x i1> %m, i32 %pivot) {
  78 ; CHECK-LABEL: test_vp_select_v256f64_vv:
  79 ; CHECK:       # %bb.0:
  80 ; CHECK-NEXT:    and %s0, %s0, (32)0
  81 ; CHECK-NEXT:    lvl %s0
  82 ; CHECK-NEXT:    vmrg %v1, %v1, %v0, %vm1
  83 ; CHECK-NEXT:    lea %s16, 256
  84 ; CHECK-NEXT:    lvl %s16
  85 ; CHECK-NEXT:    vor %v0, (0)1, %v1
  86 ; CHECK-NEXT:    b.l.t (, %s10)
  87   %r0 = call <256 x double> @llvm.vp.select.v256f64(<256 x i1> %m, <256 x double> %i0, <256 x double> %i1, i32 %pivot)
  88   ret <256 x double> %r0
  89 }
  90
  91 define fastcc <256 x double> @test_vp_select_v256f64_vr(<256 x double> %i0, double %s1, <256 x i1> %m, i32 %pivot) {
  92 ; CHECK-LABEL: test_vp_select_v256f64_vr:
  93 ; CHECK:       # %bb.0:
  94 ; CHECK-NEXT:    and %s1, %s1, (32)0
  95 ; CHECK-NEXT:    lea %s2, 256
  96 ; CHECK-NEXT:    lvl %s2
  97 ; CHECK-NEXT:    vbrd %v1, %s0
  98 ; CHECK-NEXT:    lvl %s1
  99 ; CHECK-NEXT:    vmrg %v1, %v1, %v0, %vm1
 100 ; CHECK-NEXT:    lea %s16, 256
 101 ; CHECK-NEXT:    lvl %s16
 102 ; CHECK-NEXT:    vor %v0, (0)1, %v1
 103 ; CHECK-NEXT:    b.l.t (, %s10)
 104   %xins = insertelement <256 x double> undef, double %s1, i32 0
 105   %i1 = shufflevector <256 x double> %xins, <256 x double> undef, <256 x i32> zeroinitializer
 106   %r0 = call <256 x double> @llvm.vp.select.v256f64(<256 x i1> %m, <256 x double> %i0, <256 x double> %i1, i32 %pivot)
 107   ret <256 x double> %r0
 108 }
 109
 110 declare <256 x i64> @llvm.vp.select.v256i64(<256 x i1>, <256 x i64>, <256 x i64>, i32)
 111
 112 define fastcc <256 x i64> @test_vp_select_v256i64_vv(<256 x i64> %i0, <256 x i64> %i1, <256 x i1> %m, i32 %pivot) {
 113 ; CHECK-LABEL: test_vp_select_v256i64_vv:
 114 ; CHECK:       # %bb.0:
 115 ; CHECK-NEXT:    and %s0, %s0, (32)0
 116 ; CHECK-NEXT:    lvl %s0
 117 ; CHECK-NEXT:    vmrg %v1, %v1, %v0, %vm1
 118 ; CHECK-NEXT:    lea %s16, 256
 119 ; CHECK-NEXT:    lvl %s16
 120 ; CHECK-NEXT:    vor %v0, (0)1, %v1
 121 ; CHECK-NEXT:    b.l.t (, %s10)
 122   %r0 = call <256 x i64> @llvm.vp.select.v256i64(<256 x i1> %m, <256 x i64> %i0, <256 x i64> %i1, i32 %pivot)
 123   ret <256 x i64> %r0
 124 }
 125
 126 define fastcc <256 x i64> @test_vp_select_v256i64_vr(<256 x i64> %i0, i64 %s1, <256 x i1> %m, i32 %pivot) {
 127 ; CHECK-LABEL: test_vp_select_v256i64_vr:
 128 ; CHECK:       # %bb.0:
 129 ; CHECK-NEXT:    and %s1, %s1, (32)0
 130 ; CHECK-NEXT:    lea %s2, 256
 131 ; CHECK-NEXT:    lvl %s2
 132 ; CHECK-NEXT:    vbrd %v1, %s0
 133 ; CHECK-NEXT:    lvl %s1
 134 ; CHECK-NEXT:    vmrg %v1, %v1, %v0, %vm1
 135 ; CHECK-NEXT:    lea %s16, 256
 136 ; CHECK-NEXT:    lvl %s16
 137 ; CHECK-NEXT:    vor %v0, (0)1, %v1
 138 ; CHECK-NEXT:    b.l.t (, %s10)
 139   %xins = insertelement <256 x i64> undef, i64 %s1, i32 0
 140   %i1 = shufflevector <256 x i64> %xins, <256 x i64> undef, <256 x i32> zeroinitializer
 141   %r0 = call <256 x i64> @llvm.vp.select.v256i64(<256 x i1> %m, <256 x i64> %i0, <256 x i64> %i1, i32 %pivot)
 142   ret <256 x i64> %r0
 143 }