llvm/test/Transforms/InstSimplify/vscale-inseltpoison.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -instsimplify -S -verify | FileCheck %s
   3
   4 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
   5 ;; Vector Operations
   6 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
   7
   8 ; insertelement
   9
  10 define <vscale x 4 x i32> @insertelement_idx_undef(<vscale x 4 x i32> %a) {
  11 ; CHECK-LABEL: @insertelement_idx_undef(
  12 ; CHECK-NEXT:    ret <vscale x 4 x i32> poison
  13 ;
  14   %r = insertelement <vscale x 4 x i32> %a, i32 5, i64 undef
  15   ret <vscale x 4 x i32> %r
  16 }
  17
  18 define <vscale x 4 x i32> @insertelement_value_undef(<vscale x 4 x i32> %a) {
  19 ; CHECK-LABEL: @insertelement_value_undef(
  20 ; CHECK-NEXT:    [[R:%.*]] = insertelement <vscale x 4 x i32> [[A:%.*]], i32 undef, i64 0
  21 ; CHECK-NEXT:    ret <vscale x 4 x i32> [[R]]
  22 ;
  23   %r = insertelement <vscale x 4 x i32> %a, i32 undef, i64 0
  24   ret <vscale x 4 x i32> %r
  25 }
  26
  27 define <vscale x 4 x i32> @insertelement_idx_maybe_out_of_bound(<vscale x 4 x i32> %a) {
  28 ; CHECK-LABEL: @insertelement_idx_maybe_out_of_bound(
  29 ; CHECK-NEXT:    [[R:%.*]] = insertelement <vscale x 4 x i32> [[A:%.*]], i32 5, i64 4
  30 ; CHECK-NEXT:    ret <vscale x 4 x i32> [[R]]
  31 ;
  32   %r = insertelement <vscale x 4 x i32> %a, i32 5, i64 4
  33   ret <vscale x 4 x i32> %r
  34 }
  35
  36 define <vscale x 4 x i32> @insertelement_idx_large_bound(<vscale x 4 x i32> %a) {
  37 ; CHECK-LABEL: @insertelement_idx_large_bound(
  38 ; CHECK-NEXT:    [[R:%.*]] = insertelement <vscale x 4 x i32> [[A:%.*]], i32 5, i64 12345
  39 ; CHECK-NEXT:    ret <vscale x 4 x i32> [[R]]
  40 ;
  41   %r = insertelement <vscale x 4 x i32> %a, i32 5, i64 12345
  42   ret <vscale x 4 x i32> %r
  43 }
  44
  45 define <vscale x 4 x i32> @insert_extract_element_same_vec_idx_1(<vscale x 4 x i32> %a) {
  46 ; CHECK-LABEL: @insert_extract_element_same_vec_idx_1(
  47 ; CHECK-NEXT:    ret <vscale x 4 x i32> [[A:%.*]]
  48 ;
  49   %v = extractelement <vscale x 4 x i32> %a, i64 1
  50   %r = insertelement <vscale x 4 x i32> %a, i32 %v, i64 1
  51   ret <vscale x 4 x i32> %r
  52 }
  53
  54 define <vscale x 4 x i32> @insertelement_inline_to_ret() {
  55 ; CHECK-LABEL: @insertelement_inline_to_ret(
  56 ; CHECK-NEXT:    ret <vscale x 4 x i32> insertelement (<vscale x 4 x i32> poison, i32 1, i32 0)
  57 ;
  58   %i = insertelement <vscale x 4 x i32> poison, i32 1, i32 0
  59   ret <vscale x 4 x i32> %i
  60 }
  61
  62 define <vscale x 4 x i32> @insertelement_shufflevector_inline_to_ret() {
  63 ; CHECK-LABEL: @insertelement_shufflevector_inline_to_ret(
  64 ; CHECK-NEXT:    ret <vscale x 4 x i32> shufflevector (<vscale x 4 x i32> insertelement (<vscale x 4 x i32> poison, i32 1, i32 0), <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer)
  65 ;
  66   %i = insertelement <vscale x 4 x i32> poison, i32 1, i32 0
  67   %i2 = shufflevector <vscale x 4 x i32> %i, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
  68   ret <vscale x 4 x i32> %i2
  69 }
  70
  71 ; extractelement
  72
  73 define i32 @extractelement_idx_undef(<vscale x 4 x i32> %a) {
  74 ; CHECK-LABEL: @extractelement_idx_undef(
  75 ; CHECK-NEXT:    ret i32 poison
  76 ;
  77   %r = extractelement <vscale x 4 x i32> %a, i64 undef
  78   ret i32 %r
  79 }
  80
  81 define i32 @extractelement_vec_undef(<vscale x 4 x i32> %a) {
  82 ; CHECK-LABEL: @extractelement_vec_undef(
  83 ; CHECK-NEXT:    ret i32 undef
  84 ;
  85   %r = extractelement <vscale x 4 x i32> undef, i64 1
  86   ret i32 %r
  87 }
  88
  89 define i32 @extractelement_idx_maybe_out_of_bound(<vscale x 4 x i32> %a) {
  90 ; CHECK-LABEL: @extractelement_idx_maybe_out_of_bound(
  91 ; CHECK-NEXT:    [[R:%.*]] = extractelement <vscale x 4 x i32> [[A:%.*]], i64 4
  92 ; CHECK-NEXT:    ret i32 [[R]]
  93 ;
  94   %r = extractelement <vscale x 4 x i32> %a, i64 4
  95   ret i32 %r
  96 }
  97 define i32 @extractelement_idx_large_bound(<vscale x 4 x i32> %a) {
  98 ; CHECK-LABEL: @extractelement_idx_large_bound(
  99 ; CHECK-NEXT:    [[R:%.*]] = extractelement <vscale x 4 x i32> [[A:%.*]], i64 12345
 100 ; CHECK-NEXT:    ret i32 [[R]]
 101 ;
 102   %r = extractelement <vscale x 4 x i32> %a, i64 12345
 103   ret i32 %r
 104 }
 105
 106 define i32 @insert_extract_element_same_vec_idx_2() {
 107 ; CHECK-LABEL: @insert_extract_element_same_vec_idx_2(
 108 ; CHECK-NEXT:    ret i32 1
 109 ;
 110   %v = insertelement <vscale x 4 x i32> poison, i32 1, i64 4
 111   %r = extractelement <vscale x 4 x i32> %v, i64 4
 112   ret i32 %r
 113 }
 114
 115 define i32 @insert_extract_element_same_vec_idx_3() {
 116 ; CHECK-LABEL: @insert_extract_element_same_vec_idx_3(
 117 ; CHECK-NEXT:    ret i32 1
 118 ;
 119   %r = extractelement <vscale x 4 x i32> insertelement (<vscale x 4 x i32> undef, i32 1, i64 4), i64 4
 120   ret i32 %r
 121 }
 122
 123 define i32 @insert_extract_element_same_vec_idx_4() {
 124 ; CHECK-LABEL: @insert_extract_element_same_vec_idx_4(
 125 ; CHECK-NEXT:    ret i32 1
 126 ;
 127   %r = extractelement <vscale x 4 x i32> insertelement (<vscale x 4 x i32> insertelement (<vscale x 4 x i32> undef, i32 1, i32 4), i32 2, i64 3), i64 4
 128   ret i32 %r
 129 }
 130
 131 ; more complicated expressions
 132
 133 define <vscale x 2 x i1> @cmp_le_smax_always_true(<vscale x 2 x i64> %x) {
 134 ; CHECK-LABEL: @cmp_le_smax_always_true(
 135 ; CHECK-NEXT:    ret <vscale x 2 x i1> shufflevector (<vscale x 2 x i1> insertelement (<vscale x 2 x i1> poison, i1 true, i32 0), <vscale x 2 x i1> poison, <vscale x 2 x i32> zeroinitializer)
 136 ;
 137   %cmp = icmp sle <vscale x 2 x i64> %x, shufflevector (<vscale x 2 x i64> insertelement (<vscale x 2 x i64> poison, i64 9223372036854775807, i32 0), <vscale x 2 x i64> poison, <vscale x 2 x i32> zeroinitializer)
 138   ret <vscale x 2 x i1> %cmp
 139 }
 140
 141 define <vscale x 4 x float> @bitcast() {
 142 ; CHECK-LABEL: @bitcast(
 143 ; CHECK-NEXT:    ret <vscale x 4 x float> bitcast (<vscale x 4 x i32> shufflevector (<vscale x 4 x i32> insertelement (<vscale x 4 x i32> poison, i32 1, i32 0), <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer) to <vscale x 4 x float>)
 144 ;
 145   %i1 = insertelement <vscale x 4 x i32> poison, i32 1, i32 0
 146   %i2 = shufflevector <vscale x 4 x i32> %i1, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
 147   %i3 = bitcast <vscale x 4 x i32> %i2 to <vscale x 4 x float>
 148   ret <vscale x 4 x float> %i3
 149 }
 150
 151 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 152 ;; Memory Access and Addressing Operations
 153 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 154
 155 ; getelementptr
 156
 157 define <vscale x 4 x i32*> @getelementptr_constant_foldable_1() {
 158 ; CHECK-LABEL: @getelementptr_constant_foldable_1(
 159 ; CHECK-NEXT:    ret <vscale x 4 x i32*> zeroinitializer
 160 ;
 161   %ptr = getelementptr i32, <vscale x 4 x i32*> zeroinitializer, <vscale x 4 x i64> undef
 162   ret <vscale x 4 x i32*> %ptr
 163 }
 164
 165 define <vscale x 4 x <vscale x 4 x i32>*> @getelementptr_constant_foldable_2() {
 166 ; CHECK-LABEL: @getelementptr_constant_foldable_2(
 167 ; CHECK-NEXT:    ret <vscale x 4 x <vscale x 4 x i32>*> zeroinitializer
 168 ;
 169   %ptr = getelementptr <vscale x 4 x i32>, <vscale x 4 x i32>* null, <vscale x 4 x i64> undef
 170   ret <vscale x 4 x <vscale x 4 x i32>*> %ptr
 171 }
 172
 173 ; fold getelementptr P, 0 -> P.
 174 define <vscale x 4 x i32>* @getelementptr_constant_foldable_3() {
 175 ; CHECK-LABEL: @getelementptr_constant_foldable_3(
 176 ; CHECK-NEXT:    ret <vscale x 4 x i32>* null
 177 ;
 178   %ptr = getelementptr <vscale x 4 x i32>, <vscale x 4 x i32>* null, i64 0
 179   ret <vscale x 4 x i32>* %ptr
 180 }
 181
 182 define <vscale x 4 x i32>* @getelementptr_not_constant_foldable(i64 %x) {
 183 ; CHECK-LABEL: @getelementptr_not_constant_foldable(
 184 ; CHECK-NEXT:    [[PTR:%.*]] = getelementptr <vscale x 4 x i32>, <vscale x 4 x i32>* null, i64 [[X:%.*]]
 185 ; CHECK-NEXT:    ret <vscale x 4 x i32>* [[PTR]]
 186 ;
 187   %ptr = getelementptr <vscale x 4 x i32>, <vscale x 4 x i32>* null, i64 %x
 188   ret <vscale x 4 x i32>* %ptr
 189 }
 190
 191 ; Check GEP's result is known to be non-null.
 192 define i1 @getelementptr_check_non_null(<vscale x 16 x i8>* %ptr) {
 193 ; CHECK-LABEL: @getelementptr_check_non_null(
 194 ; CHECK-NEXT:    ret i1 false
 195 ;
 196   %x = getelementptr inbounds <vscale x 16 x i8>, <vscale x 16 x i8>* %ptr, i32 1
 197   %cmp = icmp eq <vscale x 16 x i8>* %x, null
 198   ret i1 %cmp
 199 }
 200
 201 define i32 @extractelement_splat_constant_index(i32 %v) {
 202 ; CHECK-LABEL: @extractelement_splat_constant_index(
 203 ; CHECK-NEXT:    ret i32 [[V:%.*]]
 204 ;
 205   %in = insertelement <vscale x 4 x i32> poison, i32 %v, i32 0
 206   %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
 207   %r = extractelement <vscale x 4 x i32> %splat, i32 1
 208   ret i32 %r
 209 }
 210
 211 define i32 @extractelement_splat_variable_index(i32 %v, i32 %idx) {
 212 ; CHECK-LABEL: @extractelement_splat_variable_index(
 213 ; CHECK-NEXT:    ret i32 [[V:%.*]]
 214 ;
 215   %in = insertelement <vscale x 4 x i32> poison, i32 %v, i32 0
 216   %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
 217   %r = extractelement <vscale x 4 x i32> %splat, i32 %idx
 218   ret i32 %r
 219 }