250 lines · plain
1; Test vector extraction.2;3; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z13 | FileCheck %s4 5; Test v16i8 extraction of the first element.6define i8 @f1(<16 x i8> %val) {7; CHECK-LABEL: f1:8; CHECK: vlgvb %r2, %v24, 09; CHECK: br %r1410 %ret = extractelement <16 x i8> %val, i32 011 ret i8 %ret12}13 14; Test v16i8 extraction of the last element.15define i8 @f2(<16 x i8> %val) {16; CHECK-LABEL: f2:17; CHECK: vlgvb %r2, %v24, 1518; CHECK: br %r1419 %ret = extractelement <16 x i8> %val, i32 1520 ret i8 %ret21}22 23; Test v16i8 extractions of an absurd element number. This must compile24; but we don't care what it does.25define i8 @f3(<16 x i8> %val) {26; CHECK-LABEL: f3:27; CHECK-NOT: vlgvb %r2, %v24, 10000028; CHECK: br %r1429 %ret = extractelement <16 x i8> %val, i32 10000030 ret i8 %ret31}32 33; Test v16i8 extraction of a variable element.34define i8 @f4(<16 x i8> %val, i32 %index) {35; CHECK-LABEL: f4:36; CHECK: vlgvb %r2, %v24, 0(%r2)37; CHECK: br %r1438 %ret = extractelement <16 x i8> %val, i32 %index39 ret i8 %ret40}41 42; Test v8i16 extraction of the first element.43define i16 @f5(<8 x i16> %val) {44; CHECK-LABEL: f5:45; CHECK: vlgvh %r2, %v24, 046; CHECK: br %r1447 %ret = extractelement <8 x i16> %val, i32 048 ret i16 %ret49}50 51; Test v8i16 extraction of the last element.52define i16 @f6(<8 x i16> %val) {53; CHECK-LABEL: f6:54; CHECK: vlgvh %r2, %v24, 755; CHECK: br %r1456 %ret = extractelement <8 x i16> %val, i32 757 ret i16 %ret58}59 60; Test v8i16 extractions of an absurd element number. This must compile61; but we don't care what it does.62define i16 @f7(<8 x i16> %val) {63; CHECK-LABEL: f7:64; CHECK-NOT: vlgvh %r2, %v24, 10000065; CHECK: br %r1466 %ret = extractelement <8 x i16> %val, i32 10000067 ret i16 %ret68}69 70; Test v8i16 extraction of a variable element.71define i16 @f8(<8 x i16> %val, i32 %index) {72; CHECK-LABEL: f8:73; CHECK: vlgvh %r2, %v24, 0(%r2)74; CHECK: br %r1475 %ret = extractelement <8 x i16> %val, i32 %index76 ret i16 %ret77}78 79; Test v4i32 extraction of the first element.80define i32 @f9(<4 x i32> %val) {81; CHECK-LABEL: f9:82; CHECK: vlgvf %r2, %v24, 083; CHECK: br %r1484 %ret = extractelement <4 x i32> %val, i32 085 ret i32 %ret86}87 88; Test v4i32 extraction of the last element.89define i32 @f10(<4 x i32> %val) {90; CHECK-LABEL: f10:91; CHECK: vlgvf %r2, %v24, 392; CHECK: br %r1493 %ret = extractelement <4 x i32> %val, i32 394 ret i32 %ret95}96 97; Test v4i32 extractions of an absurd element number. This must compile98; but we don't care what it does.99define i32 @f11(<4 x i32> %val) {100; CHECK-LABEL: f11:101; CHECK-NOT: vlgvf %r2, %v24, 100000102; CHECK: br %r14103 %ret = extractelement <4 x i32> %val, i32 100000104 ret i32 %ret105}106 107; Test v4i32 extraction of a variable element.108define i32 @f12(<4 x i32> %val, i32 %index) {109; CHECK-LABEL: f12:110; CHECK: vlgvf %r2, %v24, 0(%r2)111; CHECK: br %r14112 %ret = extractelement <4 x i32> %val, i32 %index113 ret i32 %ret114}115 116; Test v2i64 extraction of the first element.117define i64 @f13(<2 x i64> %val) {118; CHECK-LABEL: f13:119; CHECK: vlgvg %r2, %v24, 0120; CHECK: br %r14121 %ret = extractelement <2 x i64> %val, i32 0122 ret i64 %ret123}124 125; Test v2i64 extraction of the last element.126define i64 @f14(<2 x i64> %val) {127; CHECK-LABEL: f14:128; CHECK: vlgvg %r2, %v24, 1129; CHECK: br %r14130 %ret = extractelement <2 x i64> %val, i32 1131 ret i64 %ret132}133 134; Test v2i64 extractions of an absurd element number. This must compile135; but we don't care what it does.136define i64 @f15(<2 x i64> %val) {137; CHECK-LABEL: f15:138; CHECK-NOT: vlgvg %r2, %v24, 100000139; CHECK: br %r14140 %ret = extractelement <2 x i64> %val, i32 100000141 ret i64 %ret142}143 144; Test v2i64 extraction of a variable element.145define i64 @f16(<2 x i64> %val, i32 %index) {146; CHECK-LABEL: f16:147; CHECK: vlgvg %r2, %v24, 0(%r2)148; CHECK: br %r14149 %ret = extractelement <2 x i64> %val, i32 %index150 ret i64 %ret151}152 153; Test v4f32 extraction of element 0.154define float @f17(<4 x float> %val) {155; CHECK-LABEL: f17:156; CHECK: vlr %v0, %v24157; CHECK: br %r14158 %ret = extractelement <4 x float> %val, i32 0159 ret float %ret160}161 162; Test v4f32 extraction of element 1.163define float @f18(<4 x float> %val) {164; CHECK-LABEL: f18:165; CHECK: vrepf %v0, %v24, 1166; CHECK: br %r14167 %ret = extractelement <4 x float> %val, i32 1168 ret float %ret169}170 171; Test v4f32 extraction of element 2.172define float @f19(<4 x float> %val) {173; CHECK-LABEL: f19:174; CHECK: vrepf %v0, %v24, 2175; CHECK: br %r14176 %ret = extractelement <4 x float> %val, i32 2177 ret float %ret178}179 180; Test v4f32 extraction of element 3.181define float @f20(<4 x float> %val) {182; CHECK-LABEL: f20:183; CHECK: vrepf %v0, %v24, 3184; CHECK: br %r14185 %ret = extractelement <4 x float> %val, i32 3186 ret float %ret187}188 189; Test v4f32 extractions of an absurd element number. This must compile190; but we don't care what it does.191define float @f21(<4 x float> %val) {192 %ret = extractelement <4 x float> %val, i32 100000193 ret float %ret194}195 196; Test v4f32 extraction of a variable element.197define float @f22(<4 x float> %val, i32 %index) {198; CHECK-LABEL: f22:199; CHECK: vlgvf [[REG:%r[0-5]]], %v24, 0(%r2)200; CHECK: vlvgf %v0, [[REG]], 0201; CHECK: br %r14202 %ret = extractelement <4 x float> %val, i32 %index203 ret float %ret204}205 206; Test v2f64 extraction of the first element.207define double @f23(<2 x double> %val) {208; CHECK-LABEL: f23:209; CHECK: vlr %v0, %v24210; CHECK: br %r14211 %ret = extractelement <2 x double> %val, i32 0212 ret double %ret213}214 215; Test v2f64 extraction of the last element.216define double @f24(<2 x double> %val) {217; CHECK-LABEL: f24:218; CHECK: vrepg %v0, %v24, 1219; CHECK: br %r14220 %ret = extractelement <2 x double> %val, i32 1221 ret double %ret222}223 224; Test v2f64 extractions of an absurd element number. This must compile225; but we don't care what it does.226define double @f25(<2 x double> %val) {227 %ret = extractelement <2 x double> %val, i32 100000228 ret double %ret229}230 231; Test v2f64 extraction of a variable element.232define double @f26(<2 x double> %val, i32 %index) {233; CHECK-LABEL: f26:234; CHECK: vlgvg [[REG:%r[0-5]]], %v24, 0(%r2)235; CHECK: ldgr %f0, [[REG]]236; CHECK: br %r14237 %ret = extractelement <2 x double> %val, i32 %index238 ret double %ret239}240 241; Test v16i8 extraction of a variable element with an offset.242define i8 @f27(<16 x i8> %val, i32 %index) {243; CHECK-LABEL: f27:244; CHECK: vlgvb %r2, %v24, 1(%r2)245; CHECK: br %r14246 %add = add i32 %index, 1247 %ret = extractelement <16 x i8> %val, i32 %add248 ret i8 %ret249}250