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

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes=instsimplify,verify -S | 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, i64 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 define i32 @extractelement_splat_constant_index(i32 %v) {
 156 ; CHECK-LABEL: @extractelement_splat_constant_index(
 157 ; CHECK-NEXT:    ret i32 [[V:%.*]]
 158 ;
 159   %in = insertelement <vscale x 4 x i32> poison, i32 %v, i32 0
 160   %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
 161   %r = extractelement <vscale x 4 x i32> %splat, i32 1
 162   ret i32 %r
 163 }
 164
 165 define i32 @extractelement_splat_variable_index(i32 %v, i32 %idx) {
 166 ; CHECK-LABEL: @extractelement_splat_variable_index(
 167 ; CHECK-NEXT:    ret i32 [[V:%.*]]
 168 ;
 169   %in = insertelement <vscale x 4 x i32> poison, i32 %v, i32 0
 170   %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
 171   %r = extractelement <vscale x 4 x i32> %splat, i32 %idx
 172   ret i32 %r
 173 }