1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py 2; RUN: opt < %s -passes=instcombine -S | FileCheck %s 3 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 1 11; CHECK-NEXT: ret i32 [[R]] 12; 13 %r = extractelement <vscale x 4 x i32> %a, i64 1 14 ret i32 %r 15} 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 4 20; CHECK-NEXT: ret i32 [[R]] 21; 22 %r = extractelement <vscale x 4 x i32> %a, i64 4 23 ret i32 %r 24} 25 26define i32 @extractelement_bitcast(float %f) { 27; CHECK-LABEL: @extractelement_bitcast( 28; CHECK-NEXT: [[R:%.*]] = bitcast float [[F:%.*]] to i32 29; CHECK-NEXT: ret i32 [[R]] 30; 31 %vec_float = insertelement <vscale x 4 x float> poison, float %f, i32 0 32 %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 0 34 ret i32 %r 35} 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 i8 40; CHECK-NEXT: ret i8 [[R]] 41; 42 %vec = insertelement <vscale x 2 x i32> %a, i32 %x, i32 1 43 %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 4 45 ret i8 %r 46} 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 2 52; CHECK-NEXT: ret i8 [[R]] 53; 54 %vec = insertelement <vscale x 2 x i32> %a, i32 %x, i32 1 55 %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 2 57 ret i8 %r 58} 59 60; extra use tests 61 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 1 65; 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 2 68; CHECK-NEXT: ret i8 [[R]] 69; 70 %vec = insertelement <vscale x 2 x i32> %a, i32 %x, i32 1 71 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 2 74 ret i8 %r 75} 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 1 80; 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 2 83; CHECK-NEXT: ret i8 [[R]] 84; 85 %vec = insertelement <vscale x 2 x i32> %a, i32 %x, i32 1 86 %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 2 89 ret i8 %r 90} 91 92define i32 @extractelement_shuffle_maybe_out_of_range(i32 %v) { 93; CHECK-LABEL: @extractelement_shuffle_maybe_out_of_range( 94; CHECK-NEXT: [[IN:%.*]] = insertelement <vscale x 4 x i32> poison, i32 [[V:%.*]], i64 0 95; CHECK-NEXT: [[SPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[IN]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer 96; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 4 x i32> [[SPLAT]], i64 4 97; CHECK-NEXT: ret i32 [[R]] 98; 99 %in = insertelement <vscale x 4 x i32> poison, i32 %v, i32 0 100 %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer 101 %r = extractelement <vscale x 4 x i32> %splat, i32 4 102 ret i32 %r 103} 104 105define i32 @extractelement_shuffle_invalid_index(i32 %v) { 106; CHECK-LABEL: @extractelement_shuffle_invalid_index( 107; CHECK-NEXT: [[IN:%.*]] = insertelement <vscale x 4 x i32> poison, i32 [[V:%.*]], i64 0 108; CHECK-NEXT: [[SPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[IN]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer 109; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 4 x i32> [[SPLAT]], i64 4294967295 110; CHECK-NEXT: ret i32 [[R]] 111; 112 %in = insertelement <vscale x 4 x i32> poison, i32 %v, i32 0 113 %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer 114 %r = extractelement <vscale x 4 x i32> %splat, i32 -1 115 ret i32 %r 116} 117 118define <vscale x 4 x i32> @extractelement_insertelement_same_positions(<vscale x 4 x i32> %vec) { 119; CHECK-LABEL: @extractelement_insertelement_same_positions( 120; CHECK-NEXT: ret <vscale x 4 x i32> [[VEC:%.*]] 121; 122 %vec.e0 = extractelement <vscale x 4 x i32> %vec, i32 0 123 %vec.e1 = extractelement <vscale x 4 x i32> %vec, i32 1 124 %vec.e2 = extractelement <vscale x 4 x i32> %vec, i32 2 125 %vec.e3 = extractelement <vscale x 4 x i32> %vec, i32 3 126 %1 = insertelement <vscale x 4 x i32> %vec, i32 %vec.e0, i32 0 127 %2 = insertelement <vscale x 4 x i32> %1, i32 %vec.e1, i32 1 128 %3 = insertelement <vscale x 4 x i32> %2, i32 %vec.e2, i32 2 129 %4 = insertelement <vscale x 4 x i32> %3, i32 %vec.e3, i32 3 130 ret <vscale x 4 x i32> %4 131} 132 133define <vscale x 4 x i32> @extractelement_insertelement_diff_positions(<vscale x 4 x i32> %vec) { 134; CHECK-LABEL: @extractelement_insertelement_diff_positions( 135; CHECK-NEXT: [[VEC_E0:%.*]] = extractelement <vscale x 4 x i32> [[VEC:%.*]], i64 4 136; CHECK-NEXT: [[VEC_E1:%.*]] = extractelement <vscale x 4 x i32> [[VEC]], i64 5 137; CHECK-NEXT: [[VEC_E2:%.*]] = extractelement <vscale x 4 x i32> [[VEC]], i64 6 138; CHECK-NEXT: [[VEC_E3:%.*]] = extractelement <vscale x 4 x i32> [[VEC]], i64 7 139; CHECK-NEXT: [[TMP1:%.*]] = insertelement <vscale x 4 x i32> [[VEC]], i32 [[VEC_E0]], i64 0 140; CHECK-NEXT: [[TMP2:%.*]] = insertelement <vscale x 4 x i32> [[TMP1]], i32 [[VEC_E1]], i64 1 141; CHECK-NEXT: [[TMP3:%.*]] = insertelement <vscale x 4 x i32> [[TMP2]], i32 [[VEC_E2]], i64 2 142; CHECK-NEXT: [[TMP4:%.*]] = insertelement <vscale x 4 x i32> [[TMP3]], i32 [[VEC_E3]], i64 3 143; CHECK-NEXT: ret <vscale x 4 x i32> [[TMP4]] 144; 145 %vec.e0 = extractelement <vscale x 4 x i32> %vec, i32 4 146 %vec.e1 = extractelement <vscale x 4 x i32> %vec, i32 5 147 %vec.e2 = extractelement <vscale x 4 x i32> %vec, i32 6 148 %vec.e3 = extractelement <vscale x 4 x i32> %vec, i32 7 149 %1 = insertelement <vscale x 4 x i32> %vec, i32 %vec.e0, i32 0 150 %2 = insertelement <vscale x 4 x i32> %1, i32 %vec.e1, i32 1 151 %3 = insertelement <vscale x 4 x i32> %2, i32 %vec.e2, i32 2 152 %4 = insertelement <vscale x 4 x i32> %3, i32 %vec.e3, i32 3 153 ret <vscale x 4 x i32> %4 154} 155 156define i32 @bitcast_of_extractelement( <vscale x 2 x float> %d) { 157; CHECK-LABEL: @bitcast_of_extractelement( 158; CHECK-NEXT: [[BC:%.*]] = bitcast <vscale x 2 x float> [[D:%.*]] to <vscale x 2 x i32> 159; CHECK-NEXT: [[CAST:%.*]] = extractelement <vscale x 2 x i32> [[BC]], i64 0 160; CHECK-NEXT: ret i32 [[CAST]] 161; 162 %ext = extractelement <vscale x 2 x float> %d, i32 0 163 %cast = bitcast float %ext to i32 164 ret i32 %cast 165} 166 167define i1 @extractelement_is_zero(<vscale x 2 x i32> %d, i1 %b, i32 %z) { 168; CHECK-LABEL: @extractelement_is_zero( 169; CHECK-NEXT: [[EXT:%.*]] = extractelement <vscale x 2 x i32> [[D:%.*]], i64 0 170; CHECK-NEXT: [[BB:%.*]] = icmp eq i32 [[EXT]], 0 171; CHECK-NEXT: ret i1 [[BB]] 172; 173 %ext = extractelement <vscale x 2 x i32> %d, i32 0 174 %bb = icmp eq i32 %ext, 0 175 ret i1 %bb 176} 177 178; OSS-Fuzz #25272 179; https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=25272 180define i32 @ossfuzz_25272(float %f) { 181; CHECK-LABEL: @ossfuzz_25272( 182; CHECK-NEXT: [[VEC_FLOAT:%.*]] = insertelement <vscale x 4 x float> poison, float [[F:%.*]], i64 0 183; CHECK-NEXT: [[VEC_INT:%.*]] = bitcast <vscale x 4 x float> [[VEC_FLOAT]] to <vscale x 4 x i32> 184; CHECK-NEXT: [[E:%.*]] = extractelement <vscale x 4 x i32> [[VEC_INT]], i64 2147483647 185; CHECK-NEXT: ret i32 [[E]] 186; 187 %vec_float = insertelement <vscale x 4 x float> poison, float %f, i32 0 188 %vec_int = bitcast <vscale x 4 x float> %vec_float to <vscale x 4 x i32> 189 %E = extractelement <vscale x 4 x i32> %vec_int, i32 2147483647 190 ret i32 %E 191} 192