llvm/test/CodeGen/ARM/vget_lane.ll

   1 ; RUN: llc < %s -mattr=+neon | FileCheck %s
   2 target datalayout = "e-p:32:32:32-i1:8:32-i8:8:32-i16:16:32-i32:32:32-i64:32:32-f32:32:32-f64:32:32-v64:64:64-v128:128:128-a0:0:32"
   3 target triple = "thumbv7-elf"
   4
   5 define i32 @vget_lanes8(ptr %A) nounwind {
   6 ;CHECK-LABEL: vget_lanes8:
   7 ;CHECK: vmov.s8
   8         %tmp1 = load <8 x i8>, ptr %A
   9         %tmp2 = extractelement <8 x i8> %tmp1, i32 1
  10         %tmp3 = sext i8 %tmp2 to i32
  11         ret i32 %tmp3
  12 }
  13
  14 define i32 @vget_lanes16(ptr %A) nounwind {
  15 ;CHECK-LABEL: vget_lanes16:
  16 ;CHECK: vmov.s16
  17         %tmp1 = load <4 x i16>, ptr %A
  18         %tmp2 = extractelement <4 x i16> %tmp1, i32 1
  19         %tmp3 = sext i16 %tmp2 to i32
  20         ret i32 %tmp3
  21 }
  22
  23 define i32 @vget_laneu8(ptr %A) nounwind {
  24 ;CHECK-LABEL: vget_laneu8:
  25 ;CHECK: vmov.u8
  26         %tmp1 = load <8 x i8>, ptr %A
  27         %tmp2 = extractelement <8 x i8> %tmp1, i32 1
  28         %tmp3 = zext i8 %tmp2 to i32
  29         ret i32 %tmp3
  30 }
  31
  32 define i32 @vget_laneu16(ptr %A) nounwind {
  33 ;CHECK-LABEL: vget_laneu16:
  34 ;CHECK: vmov.u16
  35         %tmp1 = load <4 x i16>, ptr %A
  36         %tmp2 = extractelement <4 x i16> %tmp1, i32 1
  37         %tmp3 = zext i16 %tmp2 to i32
  38         ret i32 %tmp3
  39 }
  40
  41 ; Do a vector add to keep the extraction from being done directly from memory.
  42 define i32 @vget_lanei32(ptr %A) nounwind {
  43 ;CHECK-LABEL: vget_lanei32:
  44 ;CHECK: vmov.32
  45         %tmp1 = load <2 x i32>, ptr %A
  46         %tmp2 = add <2 x i32> %tmp1, %tmp1
  47         %tmp3 = extractelement <2 x i32> %tmp2, i32 1
  48         ret i32 %tmp3
  49 }
  50
  51 define i32 @vgetQ_lanes8(ptr %A) nounwind {
  52 ;CHECK-LABEL: vgetQ_lanes8:
  53 ;CHECK: vmov.s8
  54         %tmp1 = load <16 x i8>, ptr %A
  55         %tmp2 = extractelement <16 x i8> %tmp1, i32 1
  56         %tmp3 = sext i8 %tmp2 to i32
  57         ret i32 %tmp3
  58 }
  59
  60 define i32 @vgetQ_lanes16(ptr %A) nounwind {
  61 ;CHECK-LABEL: vgetQ_lanes16:
  62 ;CHECK: vmov.s16
  63         %tmp1 = load <8 x i16>, ptr %A
  64         %tmp2 = extractelement <8 x i16> %tmp1, i32 1
  65         %tmp3 = sext i16 %tmp2 to i32
  66         ret i32 %tmp3
  67 }
  68
  69 define i32 @vgetQ_laneu8(ptr %A) nounwind {
  70 ;CHECK-LABEL: vgetQ_laneu8:
  71 ;CHECK: vmov.u8
  72         %tmp1 = load <16 x i8>, ptr %A
  73         %tmp2 = extractelement <16 x i8> %tmp1, i32 1
  74         %tmp3 = zext i8 %tmp2 to i32
  75         ret i32 %tmp3
  76 }
  77
  78 define i32 @vgetQ_laneu16(ptr %A) nounwind {
  79 ;CHECK-LABEL: vgetQ_laneu16:
  80 ;CHECK: vmov.u16
  81         %tmp1 = load <8 x i16>, ptr %A
  82         %tmp2 = extractelement <8 x i16> %tmp1, i32 1
  83         %tmp3 = zext i16 %tmp2 to i32
  84         ret i32 %tmp3
  85 }
  86
  87 ; Do a vector add to keep the extraction from being done directly from memory.
  88 define i32 @vgetQ_lanei32(ptr %A) nounwind {
  89 ;CHECK-LABEL: vgetQ_lanei32:
  90 ;CHECK: vmov.32
  91         %tmp1 = load <4 x i32>, ptr %A
  92         %tmp2 = add <4 x i32> %tmp1, %tmp1
  93         %tmp3 = extractelement <4 x i32> %tmp2, i32 1
  94         ret i32 %tmp3
  95 }
  96
  97 define arm_aapcs_vfpcc void @test_vget_laneu16() nounwind {
  98 entry:
  99 ; CHECK: vmov.u16 r0, d{{.*}}[1]
 100   %arg0_uint16x4_t = alloca <4 x i16>             ; <ptr> [#uses=1]
 101   %out_uint16_t = alloca i16                      ; <ptr> [#uses=1]
 102   %"alloca point" = bitcast i32 0 to i32          ; <i32> [#uses=0]
 103   %0 = load <4 x i16>, ptr %arg0_uint16x4_t, align 8  ; <<4 x i16>> [#uses=1]
 104   %1 = extractelement <4 x i16> %0, i32 1         ; <i16> [#uses=1]
 105   %2 = add i16 %1, %1
 106   store i16 %2, ptr %out_uint16_t, align 2
 107   br label %return
 108
 109 return:                                           ; preds = %entry
 110   ret void
 111 }
 112
 113 define arm_aapcs_vfpcc void @test_vget_laneu8() nounwind {
 114 entry:
 115 ; CHECK: vmov.u8 r0, d{{.*}}[1]
 116   %arg0_uint8x8_t = alloca <8 x i8>               ; <ptr> [#uses=1]
 117   %out_uint8_t = alloca i8                        ; <ptr> [#uses=1]
 118   %"alloca point" = bitcast i32 0 to i32          ; <i32> [#uses=0]
 119   %0 = load <8 x i8>, ptr %arg0_uint8x8_t, align 8    ; <<8 x i8>> [#uses=1]
 120   %1 = extractelement <8 x i8> %0, i32 1          ; <i8> [#uses=1]
 121   %2 = add i8 %1, %1
 122   store i8 %2, ptr %out_uint8_t, align 1
 123   br label %return
 124
 125 return:                                           ; preds = %entry
 126   ret void
 127 }
 128
 129 define arm_aapcs_vfpcc void @test_vgetQ_laneu16() nounwind {
 130 entry:
 131 ; CHECK: vmov.u16 r0, d{{.*}}[1]
 132   %arg0_uint16x8_t = alloca <8 x i16>             ; <ptr> [#uses=1]
 133   %out_uint16_t = alloca i16                      ; <ptr> [#uses=1]
 134   %"alloca point" = bitcast i32 0 to i32          ; <i32> [#uses=0]
 135   %0 = load <8 x i16>, ptr %arg0_uint16x8_t, align 16 ; <<8 x i16>> [#uses=1]
 136   %1 = extractelement <8 x i16> %0, i32 1         ; <i16> [#uses=1]
 137   %2 = add i16 %1, %1
 138   store i16 %2, ptr %out_uint16_t, align 2
 139   br label %return
 140
 141 return:                                           ; preds = %entry
 142   ret void
 143 }
 144
 145 define arm_aapcs_vfpcc void @test_vgetQ_laneu8() nounwind {
 146 entry:
 147 ; CHECK: vmov.u8 r0, d{{.*}}[1]
 148   %arg0_uint8x16_t = alloca <16 x i8>             ; <ptr> [#uses=1]
 149   %out_uint8_t = alloca i8                        ; <ptr> [#uses=1]
 150   %"alloca point" = bitcast i32 0 to i32          ; <i32> [#uses=0]
 151   %0 = load <16 x i8>, ptr %arg0_uint8x16_t, align 16 ; <<16 x i8>> [#uses=1]
 152   %1 = extractelement <16 x i8> %0, i32 1         ; <i8> [#uses=1]
 153   %2 = add i8 %1, %1
 154   store i8 %2, ptr %out_uint8_t, align 1
 155   br label %return
 156
 157 return:                                           ; preds = %entry
 158   ret void
 159 }
 160
 161 define <8 x i8> @vset_lane8(ptr %A, i8 %B) nounwind {
 162 ;CHECK-LABEL: vset_lane8:
 163 ;CHECK: vmov.8
 164         %tmp1 = load <8 x i8>, ptr %A
 165         %tmp2 = insertelement <8 x i8> %tmp1, i8 %B, i32 1
 166         ret <8 x i8> %tmp2
 167 }
 168
 169 define <4 x i16> @vset_lane16(ptr %A, i16 %B) nounwind {
 170 ;CHECK-LABEL: vset_lane16:
 171 ;CHECK: vmov.16
 172         %tmp1 = load <4 x i16>, ptr %A
 173         %tmp2 = insertelement <4 x i16> %tmp1, i16 %B, i32 1
 174         ret <4 x i16> %tmp2
 175 }
 176
 177 define <2 x i32> @vset_lane32(ptr %A, i32 %B) nounwind {
 178 ;CHECK-LABEL: vset_lane32:
 179 ;CHECK: vmov.32
 180         %tmp1 = load <2 x i32>, ptr %A
 181         %tmp2 = insertelement <2 x i32> %tmp1, i32 %B, i32 1
 182         ret <2 x i32> %tmp2
 183 }
 184
 185 define <16 x i8> @vsetQ_lane8(ptr %A, i8 %B) nounwind {
 186 ;CHECK-LABEL: vsetQ_lane8:
 187 ;CHECK: vmov.8
 188         %tmp1 = load <16 x i8>, ptr %A
 189         %tmp2 = insertelement <16 x i8> %tmp1, i8 %B, i32 1
 190         ret <16 x i8> %tmp2
 191 }
 192
 193 define <8 x i16> @vsetQ_lane16(ptr %A, i16 %B) nounwind {
 194 ;CHECK-LABEL: vsetQ_lane16:
 195 ;CHECK: vmov.16
 196         %tmp1 = load <8 x i16>, ptr %A
 197         %tmp2 = insertelement <8 x i16> %tmp1, i16 %B, i32 1
 198         ret <8 x i16> %tmp2
 199 }
 200
 201 define <4 x i32> @vsetQ_lane32(ptr %A, i32 %B) nounwind {
 202 ;CHECK-LABEL: vsetQ_lane32:
 203 ;CHECK: vmov.32 d{{.*}}[1], r1
 204         %tmp1 = load <4 x i32>, ptr %A
 205         %tmp2 = insertelement <4 x i32> %tmp1, i32 %B, i32 1
 206         ret <4 x i32> %tmp2
 207 }
 208
 209 define arm_aapcs_vfpcc <2 x float> @test_vset_lanef32(float %arg0_float32_t, <2 x float> %arg1_float32x2_t) nounwind {
 210 ;CHECK-LABEL: test_vset_lanef32:
 211 ;CHECK: vmov.f32 s3, s0
 212 ;CHECK: vmov.f64 d0, d1
 213 entry:
 214   %0 = insertelement <2 x float> %arg1_float32x2_t, float %arg0_float32_t, i32 1 ; <<2 x float>> [#uses=1]
 215   ret <2 x float> %0
 216 }
 217
 218 ; The llvm extractelement instruction does not require that the lane number
 219 ; be an immediate constant.  Make sure a variable lane number is handled.
 220
 221 define i32 @vget_variable_lanes8(ptr %A, i32 %B) nounwind {
 222         %tmp1 = load <8 x i8>, ptr %A
 223         %tmp2 = extractelement <8 x i8> %tmp1, i32 %B
 224         %tmp3 = sext i8 %tmp2 to i32
 225         ret i32 %tmp3
 226 }
 227
 228 define i32 @vgetQ_variable_lanei32(ptr %A, i32 %B) nounwind {
 229         %tmp1 = load <4 x i32>, ptr %A
 230         %tmp2 = add <4 x i32> %tmp1, %tmp1
 231         %tmp3 = extractelement <4 x i32> %tmp2, i32 %B
 232         ret i32 %tmp3
 233 }