llvm/test/Transforms/InstSimplify/extract-element.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes=instsimplify -S | FileCheck %s
   3 ; RUN: opt < %s -passes=instsimplify -use-constant-fp-for-fixed-length-splat -use-constant-int-for-fixed-length-splat -S | FileCheck %s
   4
   5 ; Weird Types
   6
   7 define i129 @vec_extract_negidx(<3 x i129> %a) {
   8 ; CHECK-LABEL: @vec_extract_negidx(
   9 ; CHECK-NEXT:    ret i129 poison
  10 ;
  11   %E1 = extractelement <3 x i129> %a, i129 -1
  12   ret i129 %E1
  13 }
  14
  15 define i129 @vec_extract_out_of_bounds(<3 x i129> %a) {
  16 ; CHECK-LABEL: @vec_extract_out_of_bounds(
  17 ; CHECK-NEXT:    ret i129 poison
  18 ;
  19   %E1 = extractelement <3 x i129> %a, i129 3
  20   ret i129 %E1
  21 }
  22
  23 define i129 @vec_extract_out_of_bounds2(<3 x i129> %a) {
  24 ; CHECK-LABEL: @vec_extract_out_of_bounds2(
  25 ; CHECK-NEXT:    ret i129 poison
  26 ;
  27   %E1 = extractelement <3 x i129> %a, i129 999999999999999
  28   ret i129 %E1
  29 }
  30
  31 define i129 @vec_extract_undef_index(<3 x i129> %a) {
  32 ; CHECK-LABEL: @vec_extract_undef_index(
  33 ; CHECK-NEXT:    ret i129 poison
  34 ;
  35   %E1 = extractelement <3 x i129> %a, i129 undef
  36   ret i129 %E1
  37 }
  38
  39 define i129 @vec_extract_poison_index(<3 x i129> %a) {
  40 ; CHECK-LABEL: @vec_extract_poison_index(
  41 ; CHECK-NEXT:    ret i129 poison
  42 ;
  43   %E1 = extractelement <3 x i129> %a, i129 poison
  44   ret i129 %E1
  45 }
  46
  47 define i129 @vec_extract_in_bounds(<3 x i129> %a) {
  48 ; CHECK-LABEL: @vec_extract_in_bounds(
  49 ; CHECK-NEXT:    [[E1:%.*]] = extractelement <3 x i129> [[A:%.*]], i129 2
  50 ; CHECK-NEXT:    ret i129 [[E1]]
  51 ;
  52   %E1 = extractelement <3 x i129> %a, i129 2
  53   ret i129 %E1
  54 }
  55
  56 define float @extract_element_splat_constant_vector_variable_index(i32 %y) {
  57 ; CHECK-LABEL: @extract_element_splat_constant_vector_variable_index(
  58 ; CHECK-NEXT:    ret float 2.000000e+00
  59 ;
  60   %r = extractelement <4 x float> <float 2.0, float 2.0, float 2.0, float 2.0>, i32 %y
  61   ret float %r
  62 }
  63
  64 define i32 @extractelement_splat_variable_index(i32 %v, i32 %index) {
  65 ; CHECK-LABEL: @extractelement_splat_variable_index(
  66 ; CHECK-NEXT:    ret i32 [[V:%.*]]
  67 ;
  68   %in = insertelement <3 x i32> poison, i32 %v, i32 0
  69   %splat = shufflevector <3 x i32> %in, <3 x i32> poison, <3 x i32> zeroinitializer
  70   %r = extractelement <3 x i32> %splat, i32 %index
  71   ret i32 %r
  72 }