llvm/test/Transforms/InstCombine/vscale_extractelement-inseltpoison.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes=instcombine -S | FileCheck %s
   3
   4
   5 declare void @use_vscale_2_i32(<vscale x 2 x i32>)
   6 declare void @use_vscale_8_i8(<vscale x 8 x i8>)
   7
   8 define 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
  17 define 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
  26 define 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
  37 define 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
  48 define 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
  62 define 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
  77 define 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
  92 define 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
 105 define 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
 118 define <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
 133 define <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
 156 define 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
 167 define 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
 180 define 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 }