189 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py2; RUN: opt < %s -passes=instcombine -S | FileCheck %s3 4 5declare void @use_vscale_2_i32(<vscale x 2 x i32>)6declare void @use_vscale_8_i8(<vscale x 8 x i8>)7 8define i32 @extractelement_in_range(<vscale x 4 x i32> %a) {9; CHECK-LABEL: @extractelement_in_range(10; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 4 x i32> [[A:%.*]], i64 111; CHECK-NEXT: ret i32 [[R]]12;13 %r = extractelement <vscale x 4 x i32> %a, i64 114 ret i32 %r15}16 17define i32 @extractelement_maybe_out_of_range(<vscale x 4 x i32> %a) {18; CHECK-LABEL: @extractelement_maybe_out_of_range(19; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 4 x i32> [[A:%.*]], i64 420; CHECK-NEXT: ret i32 [[R]]21;22 %r = extractelement <vscale x 4 x i32> %a, i64 423 ret i32 %r24}25 26define i32 @extractelement_bitcast(float %f) {27; CHECK-LABEL: @extractelement_bitcast(28; CHECK-NEXT: [[R:%.*]] = bitcast float [[F:%.*]] to i3229; CHECK-NEXT: ret i32 [[R]]30;31 %vec_float = insertelement <vscale x 4 x float> poison, float %f, i32 032 %vec_int = bitcast <vscale x 4 x float> %vec_float to <vscale x 4 x i32>33 %r = extractelement <vscale x 4 x i32> %vec_int, i32 034 ret i32 %r35}36 37define i8 @extractelement_bitcast_to_trunc(<vscale x 2 x i32> %a, i32 %x) {38; CHECK-LABEL: @extractelement_bitcast_to_trunc(39; CHECK-NEXT: [[R:%.*]] = trunc i32 [[X:%.*]] to i840; CHECK-NEXT: ret i8 [[R]]41;42 %vec = insertelement <vscale x 2 x i32> %a, i32 %x, i32 143 %vec_cast = bitcast <vscale x 2 x i32> %vec to <vscale x 8 x i8>44 %r = extractelement <vscale x 8 x i8> %vec_cast, i32 445 ret i8 %r46}47 48define i8 @extractelement_bitcast_useless_insert(<vscale x 2 x i32> %a, i32 %x) {49; CHECK-LABEL: @extractelement_bitcast_useless_insert(50; CHECK-NEXT: [[TMP1:%.*]] = bitcast <vscale x 2 x i32> [[A:%.*]] to <vscale x 8 x i8>51; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 8 x i8> [[TMP1]], i64 252; CHECK-NEXT: ret i8 [[R]]53;54 %vec = insertelement <vscale x 2 x i32> %a, i32 %x, i32 155 %vec_cast = bitcast <vscale x 2 x i32> %vec to <vscale x 8 x i8>56 %r = extractelement <vscale x 8 x i8> %vec_cast, i32 257 ret i8 %r58}59 60; extra use tests61 62define i8 @extractelement_bitcast_insert_extra_use_insert(<vscale x 2 x i32> %a, i32 %x) {63; CHECK-LABEL: @extractelement_bitcast_insert_extra_use_insert(64; CHECK-NEXT: [[VEC:%.*]] = insertelement <vscale x 2 x i32> [[A:%.*]], i32 [[X:%.*]], i64 165; CHECK-NEXT: call void @use_vscale_2_i32(<vscale x 2 x i32> [[VEC]])66; CHECK-NEXT: [[VEC_CAST:%.*]] = bitcast <vscale x 2 x i32> [[VEC]] to <vscale x 8 x i8>67; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 8 x i8> [[VEC_CAST]], i64 268; CHECK-NEXT: ret i8 [[R]]69;70 %vec = insertelement <vscale x 2 x i32> %a, i32 %x, i32 171 call void @use_vscale_2_i32(<vscale x 2 x i32> %vec)72 %vec_cast = bitcast <vscale x 2 x i32> %vec to <vscale x 8 x i8>73 %r = extractelement <vscale x 8 x i8> %vec_cast, i32 274 ret i8 %r75}76 77define i8 @extractelement_bitcast_insert_extra_use_bitcast(<vscale x 2 x i32> %a, i32 %x) {78; CHECK-LABEL: @extractelement_bitcast_insert_extra_use_bitcast(79; CHECK-NEXT: [[VEC:%.*]] = insertelement <vscale x 2 x i32> [[A:%.*]], i32 [[X:%.*]], i64 180; CHECK-NEXT: [[VEC_CAST:%.*]] = bitcast <vscale x 2 x i32> [[VEC]] to <vscale x 8 x i8>81; CHECK-NEXT: call void @use_vscale_8_i8(<vscale x 8 x i8> [[VEC_CAST]])82; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 8 x i8> [[VEC_CAST]], i64 283; CHECK-NEXT: ret i8 [[R]]84;85 %vec = insertelement <vscale x 2 x i32> %a, i32 %x, i32 186 %vec_cast = bitcast <vscale x 2 x i32> %vec to <vscale x 8 x i8>87 call void @use_vscale_8_i8(<vscale x 8 x i8> %vec_cast)88 %r = extractelement <vscale x 8 x i8> %vec_cast, i32 289 ret i8 %r90}91 92; while it may be that the extract is out-of-bounds, any valid index93; is going to yield %v (because the mask is all-zeros).94 95define i32 @extractelement_shuffle_maybe_out_of_range(i32 %v) {96; CHECK-LABEL: @extractelement_shuffle_maybe_out_of_range(97; CHECK-NEXT: ret i32 [[V:%.*]]98;99 %in = insertelement <vscale x 4 x i32> poison, i32 %v, i32 0100 %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer101 %r = extractelement <vscale x 4 x i32> %splat, i32 4102 ret i32 %r103}104 105define i32 @extractelement_shuffle_invalid_index(i32 %v) {106; CHECK-LABEL: @extractelement_shuffle_invalid_index(107; CHECK-NEXT: ret i32 [[V:%.*]]108;109 %in = insertelement <vscale x 4 x i32> poison, i32 %v, i32 0110 %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer111 %r = extractelement <vscale x 4 x i32> %splat, i32 -1112 ret i32 %r113}114 115define <vscale x 4 x i32> @extractelement_insertelement_same_positions(<vscale x 4 x i32> %vec) {116; CHECK-LABEL: @extractelement_insertelement_same_positions(117; CHECK-NEXT: ret <vscale x 4 x i32> [[VEC:%.*]]118;119 %vec.e0 = extractelement <vscale x 4 x i32> %vec, i32 0120 %vec.e1 = extractelement <vscale x 4 x i32> %vec, i32 1121 %vec.e2 = extractelement <vscale x 4 x i32> %vec, i32 2122 %vec.e3 = extractelement <vscale x 4 x i32> %vec, i32 3123 %1 = insertelement <vscale x 4 x i32> %vec, i32 %vec.e0, i32 0124 %2 = insertelement <vscale x 4 x i32> %1, i32 %vec.e1, i32 1125 %3 = insertelement <vscale x 4 x i32> %2, i32 %vec.e2, i32 2126 %4 = insertelement <vscale x 4 x i32> %3, i32 %vec.e3, i32 3127 ret <vscale x 4 x i32> %4128}129 130define <vscale x 4 x i32> @extractelement_insertelement_diff_positions(<vscale x 4 x i32> %vec) {131; CHECK-LABEL: @extractelement_insertelement_diff_positions(132; CHECK-NEXT: [[VEC_E0:%.*]] = extractelement <vscale x 4 x i32> [[VEC:%.*]], i64 4133; CHECK-NEXT: [[VEC_E1:%.*]] = extractelement <vscale x 4 x i32> [[VEC]], i64 5134; CHECK-NEXT: [[VEC_E2:%.*]] = extractelement <vscale x 4 x i32> [[VEC]], i64 6135; CHECK-NEXT: [[VEC_E3:%.*]] = extractelement <vscale x 4 x i32> [[VEC]], i64 7136; CHECK-NEXT: [[TMP1:%.*]] = insertelement <vscale x 4 x i32> [[VEC]], i32 [[VEC_E0]], i64 0137; CHECK-NEXT: [[TMP2:%.*]] = insertelement <vscale x 4 x i32> [[TMP1]], i32 [[VEC_E1]], i64 1138; CHECK-NEXT: [[TMP3:%.*]] = insertelement <vscale x 4 x i32> [[TMP2]], i32 [[VEC_E2]], i64 2139; CHECK-NEXT: [[TMP4:%.*]] = insertelement <vscale x 4 x i32> [[TMP3]], i32 [[VEC_E3]], i64 3140; CHECK-NEXT: ret <vscale x 4 x i32> [[TMP4]]141;142 %vec.e0 = extractelement <vscale x 4 x i32> %vec, i32 4143 %vec.e1 = extractelement <vscale x 4 x i32> %vec, i32 5144 %vec.e2 = extractelement <vscale x 4 x i32> %vec, i32 6145 %vec.e3 = extractelement <vscale x 4 x i32> %vec, i32 7146 %1 = insertelement <vscale x 4 x i32> %vec, i32 %vec.e0, i32 0147 %2 = insertelement <vscale x 4 x i32> %1, i32 %vec.e1, i32 1148 %3 = insertelement <vscale x 4 x i32> %2, i32 %vec.e2, i32 2149 %4 = insertelement <vscale x 4 x i32> %3, i32 %vec.e3, i32 3150 ret <vscale x 4 x i32> %4151}152 153define i32 @bitcast_of_extractelement( <vscale x 2 x float> %d) {154; CHECK-LABEL: @bitcast_of_extractelement(155; CHECK-NEXT: [[BC:%.*]] = bitcast <vscale x 2 x float> [[D:%.*]] to <vscale x 2 x i32>156; CHECK-NEXT: [[CAST:%.*]] = extractelement <vscale x 2 x i32> [[BC]], i64 0157; CHECK-NEXT: ret i32 [[CAST]]158;159 %ext = extractelement <vscale x 2 x float> %d, i32 0160 %cast = bitcast float %ext to i32161 ret i32 %cast162}163 164define i1 @extractelement_is_zero(<vscale x 2 x i32> %d, i1 %b, i32 %z) {165; CHECK-LABEL: @extractelement_is_zero(166; CHECK-NEXT: [[EXT:%.*]] = extractelement <vscale x 2 x i32> [[D:%.*]], i64 0167; CHECK-NEXT: [[BB:%.*]] = icmp eq i32 [[EXT]], 0168; CHECK-NEXT: ret i1 [[BB]]169;170 %ext = extractelement <vscale x 2 x i32> %d, i32 0171 %bb = icmp eq i32 %ext, 0172 ret i1 %bb173}174 175; OSS-Fuzz #25272176; https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=25272177define i32 @ossfuzz_25272(float %f) {178; CHECK-LABEL: @ossfuzz_25272(179; CHECK-NEXT: [[VEC_FLOAT:%.*]] = insertelement <vscale x 4 x float> poison, float [[F:%.*]], i64 0180; CHECK-NEXT: [[VEC_INT:%.*]] = bitcast <vscale x 4 x float> [[VEC_FLOAT]] to <vscale x 4 x i32>181; CHECK-NEXT: [[E:%.*]] = extractelement <vscale x 4 x i32> [[VEC_INT]], i64 2147483647182; CHECK-NEXT: ret i32 [[E]]183;184 %vec_float = insertelement <vscale x 4 x float> poison, float %f, i32 0185 %vec_int = bitcast <vscale x 4 x float> %vec_float to <vscale x 4 x i32>186 %E = extractelement <vscale x 4 x i32> %vec_int, i32 2147483647187 ret i32 %E188}189