332 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py2; RUN: opt < %s -passes=instcombine -S -data-layout="e" | FileCheck %s --check-prefixes=ANY,LE3; RUN: opt < %s -passes=instcombine -S -data-layout="E" | FileCheck %s --check-prefixes=ANY,BE4 5define i32 @extractelement_out_of_range(<2 x i32> %x) {6; ANY-LABEL: @extractelement_out_of_range(7; ANY-NEXT: ret i32 poison8;9 %E1 = extractelement <2 x i32> %x, i8 1610 ret i32 %E111}12 13define i32 @extractelement_type_out_of_range(<2 x i32> %x) {14; ANY-LABEL: @extractelement_type_out_of_range(15; ANY-NEXT: [[E1:%.*]] = extractelement <2 x i32> [[X:%.*]], i64 016; ANY-NEXT: ret i32 [[E1]]17;18 %E1 = extractelement <2 x i32> %x, i128 019 ret i32 %E120}21 22define i32 @bitcasted_inselt_equal_num_elts(float %f) {23; ANY-LABEL: @bitcasted_inselt_equal_num_elts(24; ANY-NEXT: [[R:%.*]] = bitcast float [[F:%.*]] to i3225; ANY-NEXT: ret i32 [[R]]26;27 %vf = insertelement <4 x float> poison, float %f, i32 028 %vi = bitcast <4 x float> %vf to <4 x i32>29 %r = extractelement <4 x i32> %vi, i32 030 ret i32 %r31}32 33define i64 @test2(i64 %in) {34; ANY-LABEL: @test2(35; ANY-NEXT: ret i64 [[IN:%.*]]36;37 %vec = insertelement <8 x i64> poison, i64 %in, i32 038 %splat = shufflevector <8 x i64> %vec, <8 x i64> poison, <8 x i32> zeroinitializer39 %add = add <8 x i64> %splat, <i64 0, i64 1, i64 2, i64 3, i64 4, i64 5, i64 6, i64 7>40 %r = extractelement <8 x i64> %add, i32 041 ret i64 %r42}43 44define i32 @bitcasted_inselt_wide_source_zero_elt(i64 %x) {45; LE-LABEL: @bitcasted_inselt_wide_source_zero_elt(46; LE-NEXT: [[R:%.*]] = trunc i64 [[X:%.*]] to i3247; LE-NEXT: ret i32 [[R]]48;49; BE-LABEL: @bitcasted_inselt_wide_source_zero_elt(50; BE-NEXT: [[TMP1:%.*]] = lshr i64 [[X:%.*]], 3251; BE-NEXT: [[R:%.*]] = trunc nuw i64 [[TMP1]] to i3252; BE-NEXT: ret i32 [[R]]53;54 %i = insertelement <2 x i64> zeroinitializer, i64 %x, i32 055 %b = bitcast <2 x i64> %i to <4 x i32>56 %r = extractelement <4 x i32> %b, i32 057 ret i32 %r58}59 60define i16 @bitcasted_inselt_wide_source_modulo_elt(i64 %x) {61; LE-LABEL: @bitcasted_inselt_wide_source_modulo_elt(62; LE-NEXT: [[R:%.*]] = trunc i64 [[X:%.*]] to i1663; LE-NEXT: ret i16 [[R]]64;65; BE-LABEL: @bitcasted_inselt_wide_source_modulo_elt(66; BE-NEXT: [[TMP1:%.*]] = lshr i64 [[X:%.*]], 4867; BE-NEXT: [[R:%.*]] = trunc nuw i64 [[TMP1]] to i1668; BE-NEXT: ret i16 [[R]]69;70 %i = insertelement <2 x i64> poison, i64 %x, i32 171 %b = bitcast <2 x i64> %i to <8 x i16>72 %r = extractelement <8 x i16> %b, i32 473 ret i16 %r74}75 76define i32 @bitcasted_inselt_wide_source_not_modulo_elt(i64 %x) {77; LE-LABEL: @bitcasted_inselt_wide_source_not_modulo_elt(78; LE-NEXT: [[TMP1:%.*]] = lshr i64 [[X:%.*]], 3279; LE-NEXT: [[R:%.*]] = trunc nuw i64 [[TMP1]] to i3280; LE-NEXT: ret i32 [[R]]81;82; BE-LABEL: @bitcasted_inselt_wide_source_not_modulo_elt(83; BE-NEXT: [[R:%.*]] = trunc i64 [[X:%.*]] to i3284; BE-NEXT: ret i32 [[R]]85;86 %i = insertelement <2 x i64> poison, i64 %x, i32 087 %b = bitcast <2 x i64> %i to <4 x i32>88 %r = extractelement <4 x i32> %b, i32 189 ret i32 %r90}91 92define i8 @bitcasted_inselt_wide_source_not_modulo_elt_not_half(i32 %x) {93; LE-LABEL: @bitcasted_inselt_wide_source_not_modulo_elt_not_half(94; LE-NEXT: [[TMP1:%.*]] = lshr i32 [[X:%.*]], 1695; LE-NEXT: [[R:%.*]] = trunc i32 [[TMP1]] to i896; LE-NEXT: ret i8 [[R]]97;98; BE-LABEL: @bitcasted_inselt_wide_source_not_modulo_elt_not_half(99; BE-NEXT: [[TMP1:%.*]] = lshr i32 [[X:%.*]], 8100; BE-NEXT: [[R:%.*]] = trunc i32 [[TMP1]] to i8101; BE-NEXT: ret i8 [[R]]102;103 %i = insertelement <2 x i32> poison, i32 %x, i32 0104 %b = bitcast <2 x i32> %i to <8 x i8>105 %r = extractelement <8 x i8> %b, i32 2106 ret i8 %r107}108 109define i3 @bitcasted_inselt_wide_source_not_modulo_elt_not_half_weird_types(i15 %x) {110; LE-LABEL: @bitcasted_inselt_wide_source_not_modulo_elt_not_half_weird_types(111; LE-NEXT: [[TMP1:%.*]] = lshr i15 [[X:%.*]], 3112; LE-NEXT: [[R:%.*]] = trunc i15 [[TMP1]] to i3113; LE-NEXT: ret i3 [[R]]114;115; BE-LABEL: @bitcasted_inselt_wide_source_not_modulo_elt_not_half_weird_types(116; BE-NEXT: [[TMP1:%.*]] = lshr i15 [[X:%.*]], 9117; BE-NEXT: [[R:%.*]] = trunc i15 [[TMP1]] to i3118; BE-NEXT: ret i3 [[R]]119;120 %i = insertelement <3 x i15> poison, i15 %x, i32 0121 %b = bitcast <3 x i15> %i to <15 x i3>122 %r = extractelement <15 x i3> %b, i32 1123 ret i3 %r124}125 126; Negative test for the above fold, but we can remove the insert here.127 128define i8 @bitcasted_inselt_wide_source_wrong_insert(<2 x i32> %v, i32 %x) {129; ANY-LABEL: @bitcasted_inselt_wide_source_wrong_insert(130; ANY-NEXT: [[TMP1:%.*]] = bitcast <2 x i32> [[V:%.*]] to <8 x i8>131; ANY-NEXT: [[R:%.*]] = extractelement <8 x i8> [[TMP1]], i64 2132; ANY-NEXT: ret i8 [[R]]133;134 %i = insertelement <2 x i32> %v, i32 %x, i32 1135 %b = bitcast <2 x i32> %i to <8 x i8>136 %r = extractelement <8 x i8> %b, i32 2137 ret i8 %r138}139 140; Partial negative test for the above fold, extra uses are not allowed if shift is needed.141 142declare void @use(<8 x i8>)143 144define i8 @bitcasted_inselt_wide_source_uses(i32 %x) {145; LE-LABEL: @bitcasted_inselt_wide_source_uses(146; LE-NEXT: [[I:%.*]] = insertelement <2 x i32> poison, i32 [[X:%.*]], i64 0147; LE-NEXT: [[B:%.*]] = bitcast <2 x i32> [[I]] to <8 x i8>148; LE-NEXT: call void @use(<8 x i8> [[B]])149; LE-NEXT: [[R:%.*]] = extractelement <8 x i8> [[B]], i64 3150; LE-NEXT: ret i8 [[R]]151;152; BE-LABEL: @bitcasted_inselt_wide_source_uses(153; BE-NEXT: [[I:%.*]] = insertelement <2 x i32> poison, i32 [[X:%.*]], i64 0154; BE-NEXT: [[B:%.*]] = bitcast <2 x i32> [[I]] to <8 x i8>155; BE-NEXT: call void @use(<8 x i8> [[B]])156; BE-NEXT: [[R:%.*]] = trunc i32 [[X]] to i8157; BE-NEXT: ret i8 [[R]]158;159 %i = insertelement <2 x i32> poison, i32 %x, i32 0160 %b = bitcast <2 x i32> %i to <8 x i8>161 call void @use(<8 x i8> %b)162 %r = extractelement <8 x i8> %b, i32 3163 ret i8 %r164}165 166define float @bitcasted_inselt_to_FP(i64 %x) {167; LE-LABEL: @bitcasted_inselt_to_FP(168; LE-NEXT: [[TMP1:%.*]] = lshr i64 [[X:%.*]], 32169; LE-NEXT: [[TMP2:%.*]] = trunc nuw i64 [[TMP1]] to i32170; LE-NEXT: [[R:%.*]] = bitcast i32 [[TMP2]] to float171; LE-NEXT: ret float [[R]]172;173; BE-LABEL: @bitcasted_inselt_to_FP(174; BE-NEXT: [[TMP1:%.*]] = trunc i64 [[X:%.*]] to i32175; BE-NEXT: [[R:%.*]] = bitcast i32 [[TMP1]] to float176; BE-NEXT: ret float [[R]]177;178 %i = insertelement <2 x i64> poison, i64 %x, i32 0179 %b = bitcast <2 x i64> %i to <4 x float>180 %r = extractelement <4 x float> %b, i32 1181 ret float %r182}183 184declare void @use_v2i128(<2 x i128>)185declare void @use_v8f32(<8 x float>)186 187define float @bitcasted_inselt_to_FP_uses(i128 %x) {188; ANY-LABEL: @bitcasted_inselt_to_FP_uses(189; ANY-NEXT: [[I:%.*]] = insertelement <2 x i128> poison, i128 [[X:%.*]], i64 0190; ANY-NEXT: call void @use_v2i128(<2 x i128> [[I]])191; ANY-NEXT: [[B:%.*]] = bitcast <2 x i128> [[I]] to <8 x float>192; ANY-NEXT: [[R:%.*]] = extractelement <8 x float> [[B]], i64 1193; ANY-NEXT: ret float [[R]]194;195 %i = insertelement <2 x i128> poison, i128 %x, i32 0196 call void @use_v2i128(<2 x i128> %i)197 %b = bitcast <2 x i128> %i to <8 x float>198 %r = extractelement <8 x float> %b, i32 1199 ret float %r200}201 202define float @bitcasted_inselt_to_FP_uses2(i128 %x) {203; ANY-LABEL: @bitcasted_inselt_to_FP_uses2(204; ANY-NEXT: [[I:%.*]] = insertelement <2 x i128> poison, i128 [[X:%.*]], i64 0205; ANY-NEXT: [[B:%.*]] = bitcast <2 x i128> [[I]] to <8 x float>206; ANY-NEXT: call void @use_v8f32(<8 x float> [[B]])207; ANY-NEXT: [[R:%.*]] = extractelement <8 x float> [[B]], i64 1208; ANY-NEXT: ret float [[R]]209;210 %i = insertelement <2 x i128> poison, i128 %x, i32 0211 %b = bitcast <2 x i128> %i to <8 x float>212 call void @use_v8f32(<8 x float> %b)213 %r = extractelement <8 x float> %b, i32 1214 ret float %r215}216 217define i32 @bitcasted_inselt_from_FP(double %x) {218; LE-LABEL: @bitcasted_inselt_from_FP(219; LE-NEXT: [[TMP1:%.*]] = bitcast double [[X:%.*]] to i64220; LE-NEXT: [[TMP2:%.*]] = lshr i64 [[TMP1]], 32221; LE-NEXT: [[R:%.*]] = trunc nuw i64 [[TMP2]] to i32222; LE-NEXT: ret i32 [[R]]223;224; BE-LABEL: @bitcasted_inselt_from_FP(225; BE-NEXT: [[TMP1:%.*]] = bitcast double [[X:%.*]] to i64226; BE-NEXT: [[R:%.*]] = trunc i64 [[TMP1]] to i32227; BE-NEXT: ret i32 [[R]]228;229 %i = insertelement <2 x double> poison, double %x, i32 0230 %b = bitcast <2 x double> %i to <4 x i32>231 %r = extractelement <4 x i32> %b, i32 1232 ret i32 %r233}234 235declare void @use_v2f64(<2 x double>)236declare void @use_v8i16(<8 x i16>)237 238define i16 @bitcasted_inselt_from_FP_uses(double %x) {239; ANY-LABEL: @bitcasted_inselt_from_FP_uses(240; ANY-NEXT: [[I:%.*]] = insertelement <2 x double> poison, double [[X:%.*]], i64 0241; ANY-NEXT: call void @use_v2f64(<2 x double> [[I]])242; ANY-NEXT: [[B:%.*]] = bitcast <2 x double> [[I]] to <8 x i16>243; ANY-NEXT: [[R:%.*]] = extractelement <8 x i16> [[B]], i64 1244; ANY-NEXT: ret i16 [[R]]245;246 %i = insertelement <2 x double> poison, double %x, i32 0247 call void @use_v2f64(<2 x double> %i)248 %b = bitcast <2 x double> %i to <8 x i16>249 %r = extractelement <8 x i16> %b, i32 1250 ret i16 %r251}252 253define i16 @bitcasted_inselt_from_FP_uses2(double %x) {254; ANY-LABEL: @bitcasted_inselt_from_FP_uses2(255; ANY-NEXT: [[I:%.*]] = insertelement <2 x double> poison, double [[X:%.*]], i64 0256; ANY-NEXT: [[B:%.*]] = bitcast <2 x double> [[I]] to <8 x i16>257; ANY-NEXT: call void @use_v8i16(<8 x i16> [[B]])258; ANY-NEXT: [[R:%.*]] = extractelement <8 x i16> [[B]], i64 1259; ANY-NEXT: ret i16 [[R]]260;261 %i = insertelement <2 x double> poison, double %x, i32 0262 %b = bitcast <2 x double> %i to <8 x i16>263 call void @use_v8i16(<8 x i16> %b)264 %r = extractelement <8 x i16> %b, i32 1265 ret i16 %r266}267 268define float @bitcasted_inselt_to_and_from_FP(double %x) {269; ANY-LABEL: @bitcasted_inselt_to_and_from_FP(270; ANY-NEXT: [[I:%.*]] = insertelement <2 x double> poison, double [[X:%.*]], i64 0271; ANY-NEXT: [[B:%.*]] = bitcast <2 x double> [[I]] to <4 x float>272; ANY-NEXT: [[R:%.*]] = extractelement <4 x float> [[B]], i64 1273; ANY-NEXT: ret float [[R]]274;275 %i = insertelement <2 x double> poison, double %x, i32 0276 %b = bitcast <2 x double> %i to <4 x float>277 %r = extractelement <4 x float> %b, i32 1278 ret float %r279}280 281define float @bitcasted_inselt_to_and_from_FP_uses(double %x) {282; ANY-LABEL: @bitcasted_inselt_to_and_from_FP_uses(283; ANY-NEXT: [[I:%.*]] = insertelement <2 x double> poison, double [[X:%.*]], i64 0284; ANY-NEXT: call void @use_v2f64(<2 x double> [[I]])285; ANY-NEXT: [[B:%.*]] = bitcast <2 x double> [[I]] to <4 x float>286; ANY-NEXT: [[R:%.*]] = extractelement <4 x float> [[B]], i64 1287; ANY-NEXT: ret float [[R]]288;289 %i = insertelement <2 x double> poison, double %x, i32 0290 call void @use_v2f64(<2 x double> %i)291 %b = bitcast <2 x double> %i to <4 x float>292 %r = extractelement <4 x float> %b, i32 1293 ret float %r294}295 296declare void @use_v4f32(<4 x float>)297 298define float @bitcasted_inselt_to_and_from_FP_uses2(double %x) {299; ANY-LABEL: @bitcasted_inselt_to_and_from_FP_uses2(300; ANY-NEXT: [[I:%.*]] = insertelement <2 x double> poison, double [[X:%.*]], i64 0301; ANY-NEXT: [[B:%.*]] = bitcast <2 x double> [[I]] to <4 x float>302; ANY-NEXT: call void @use_v4f32(<4 x float> [[B]])303; ANY-NEXT: [[R:%.*]] = extractelement <4 x float> [[B]], i64 1304; ANY-NEXT: ret float [[R]]305;306 %i = insertelement <2 x double> poison, double %x, i32 0307 %b = bitcast <2 x double> %i to <4 x float>308 call void @use_v4f32(<4 x float> %b)309 %r = extractelement <4 x float> %b, i32 1310 ret float %r311}312 313; This would crash/assert because the logic for collectShuffleElements()314; does not consider the possibility of invalid insert/extract operands.315 316define <4 x double> @invalid_extractelement(<2 x double> %a, <4 x double> %b, ptr %p) {317; ANY-LABEL: @invalid_extractelement(318; ANY-NEXT: [[TMP1:%.*]] = shufflevector <2 x double> [[A:%.*]], <2 x double> poison, <4 x i32> <i32 0, i32 poison, i32 poison, i32 poison>319; ANY-NEXT: [[T4:%.*]] = shufflevector <4 x double> [[B:%.*]], <4 x double> [[TMP1]], <4 x i32> <i32 0, i32 1, i32 4, i32 3>320; ANY-NEXT: [[E:%.*]] = extractelement <4 x double> [[B]], i64 1321; ANY-NEXT: store double [[E]], ptr [[P:%.*]], align 8322; ANY-NEXT: ret <4 x double> [[T4]]323;324 %t3 = extractelement <2 x double> %a, i32 0325 %t4 = insertelement <4 x double> %b, double %t3, i32 2326 %e = extractelement <4 x double> %t4, i32 1327 store double %e, ptr %p328 %e1 = extractelement <2 x double> %a, i32 4 ; invalid index329 %r = insertelement <4 x double> %t4, double %e1, i64 0330 ret <4 x double> %r331}332