xref: /llvm-project/llvm/test/Transforms/InstSimplify/ConstProp/extractelement-vscale.ll (revision 56c091ea7106507b36015297ee9005c9d5fab0bf)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
2; RUN: opt -S -passes=instsimplify < %s | FileCheck %s
3
4define i8 @definitely_in_bounds() {
5; CHECK-LABEL: define i8 @definitely_in_bounds() {
6; CHECK-NEXT:    ret i8 0
7;
8  ret i8 extractelement (<vscale x 16 x i8> zeroinitializer, i64 15)
9}
10
11define i8 @maybe_in_bounds() {
12; CHECK-LABEL: define i8 @maybe_in_bounds() {
13; CHECK-NEXT:    ret i8 extractelement (<vscale x 16 x i8> zeroinitializer, i64 16)
14;
15  ret i8 extractelement (<vscale x 16 x i8> zeroinitializer, i64 16)
16}
17
18; Examples of extracting a lane from a splat constant
19
20define i32 @extractconstant_shuffle_in_range(i32 %v) {
21; CHECK-LABEL: define i32 @extractconstant_shuffle_in_range(
22; CHECK-SAME: i32 [[V:%.*]]) {
23; CHECK-NEXT:    ret i32 1024
24;
25  %in = insertelement <vscale x 4 x i32> undef, i32 1024, i32 0
26  %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
27  %r = extractelement <vscale x 4 x i32> %splat, i32 1
28  ret i32 %r
29}
30
31define i32 @extractconstant_shuffle_maybe_out_of_range(i32 %v) {
32; CHECK-LABEL: define i32 @extractconstant_shuffle_maybe_out_of_range(
33; CHECK-SAME: i32 [[V:%.*]]) {
34; CHECK-NEXT:    ret i32 extractelement (<vscale x 4 x i32> splat (i32 1024), i32 4)
35;
36  %in = insertelement <vscale x 4 x i32> undef, i32 1024, i32 0
37  %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
38  %r = extractelement <vscale x 4 x i32> %splat, i32 4
39  ret i32 %r
40}
41
42define i32 @extractconstant_shuffle_invalid_index(i32 %v) {
43; CHECK-LABEL: define i32 @extractconstant_shuffle_invalid_index(
44; CHECK-SAME: i32 [[V:%.*]]) {
45; CHECK-NEXT:    ret i32 extractelement (<vscale x 4 x i32> splat (i32 1024), i32 -1)
46;
47  %in = insertelement <vscale x 4 x i32> undef, i32 1024, i32 0
48  %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
49  %r = extractelement <vscale x 4 x i32> %splat, i32 -1
50  ret i32 %r
51}
52