xref: /llvm-project/llvm/test/Transforms/InstSimplify/extract-element.ll (revision 4872ecf1cc3cb9c4939a9e6210a9b9e9a9032e9f)
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
7define 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
15define 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
23define 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
31define 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
39define 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
47define 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
56define 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
64define 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}
73