| blob | db22b3b657fb3fb0cbc9efdc34cb0e141e5b12aa |
1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2 ; RUN: opt -S -passes=instcombine < %s | FileCheck %s
4 ; This test checks that bitcast is moved after insertelement when both vector and scalar are
5 ; bitcast from the same element type.
6 ; inselt (bitcast VecSrc), (bitcast ScalarSrc), IdxOp
7 ; --> bitcast (inselt VecSrc, ScalarSrc, IdxOp)
8 define <vscale x 4 x float> @insertelement_bitcast(<vscale x 4 x i32> %vec, i32 %x) {
9 ; CHECK-LABEL: @insertelement_bitcast(
10 ; CHECK-NEXT: [[TMP1:%.*]] = insertelement <vscale x 4 x i32> [[VEC:%.*]], i32 [[X:%.*]], i64 0
11 ; CHECK-NEXT: [[R:%.*]] = bitcast <vscale x 4 x i32> [[TMP1]] to <vscale x 4 x float>
12 ; CHECK-NEXT: ret <vscale x 4 x float> [[R]]
13 ;
14 %x_cast = bitcast i32 %x to float
15 %vec_cast = bitcast <vscale x 4 x i32> %vec to <vscale x 4 x float>
16 %r = insertelement <vscale x 4 x float> %vec_cast, float %x_cast, i32 0
17 ret <vscale x 4 x float> %r
18 }
20 ; This test checks that code-path "Try to form a shuffle from a chain of extract-insert ops" is
21 ; not taken when both extract and insert are scalable type.
22 ; For scalable type, the vector length needed to create shuffle mask is not a compile-time constant.
23 ; Meanwhile, for scalable type shuffle mask only support splat and undef in the current code base.
24 ; Otherwise we crash at:
25 ; "Assertion `isValidOperands(V1, V2, Mask) && "Invalid shuffle vector instruction operands!"' failed."
26 define <vscale x 4 x i32> @insertelement_extractelement(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
27 ; CHECK-LABEL: @insertelement_extractelement(
28 ; CHECK-NEXT: [[T0:%.*]] = extractelement <vscale x 4 x i32> [[A:%.*]], i64 1
29 ; CHECK-NEXT: [[T1:%.*]] = insertelement <vscale x 4 x i32> [[B:%.*]], i32 [[T0]], i64 0
30 ; CHECK-NEXT: ret <vscale x 4 x i32> [[T1]]
31 ;
32 %t0 = extractelement <vscale x 4 x i32> %a, i32 1
33 %t1 = insertelement <vscale x 4 x i32> %b, i32 %t0, i32 0
34 ret <vscale x 4 x i32> %t1
35 }
37 ; This test checks that we are not attempting to create a shuffle from extract/insert chain,
38 ; when extract is from a scalable type, and the insert vector is fixed-length.
39 define <4 x i32> @insertelement_extractelement_fixed_vec_extract_from_scalable(<vscale x 4 x i32> %a, <4 x i32> %b) {
40 ; CHECK-LABEL: @insertelement_extractelement_fixed_vec_extract_from_scalable(
41 ; CHECK-NEXT: [[T0:%.*]] = extractelement <vscale x 4 x i32> [[A:%.*]], i64 1
42 ; CHECK-NEXT: [[T1:%.*]] = insertelement <4 x i32> [[B:%.*]], i32 [[T0]], i64 0
43 ; CHECK-NEXT: ret <4 x i32> [[T1]]
44 ;
45 %t0 = extractelement <vscale x 4 x i32> %a, i32 1
46 %t1 = insertelement <4 x i32> %b, i32 %t0, i32 0
47 ret <4 x i32> %t1
48 }
50 ; This test checks that the optimization "foldConstantInsEltInfoShuffle" is not taken for scalable type.
51 ; Particularly the fold:
52 ; insertelt (insertelt X, C1, CIndex1), C, CIndex
53 ; --> shufflevector X, CVec', Mask'
54 ; For scalable type, the vector length needed to create shuffle mask is not a compile-time constant.
55 ; Meanwhile, for scalable type shuffle mask only support splat and undef in the current code base.
56 ; Otherwise we crash at:
57 ; "Assertion `isValidOperands(V1, V2, Mask) && "Invalid shuffle vector instruction operands!"' failed."
58 define <vscale x 4 x i32> @insertelement_insertelement(<vscale x 4 x i32> %vec) {
59 ; CHECK-LABEL: @insertelement_insertelement(
60 ; CHECK-NEXT: [[T0:%.*]] = insertelement <vscale x 4 x i32> [[VEC:%.*]], i32 1, i64 1
61 ; CHECK-NEXT: [[T1:%.*]] = insertelement <vscale x 4 x i32> [[T0]], i32 2, i64 2
62 ; CHECK-NEXT: ret <vscale x 4 x i32> [[T1]]
63 ;
64 %t0 = insertelement <vscale x 4 x i32> %vec, i32 1, i32 1
65 %t1 = insertelement <vscale x 4 x i32> %t0, i32 2, i32 2
66 ret <vscale x 4 x i32> %t1
67 }
69 ; This test checks that the following insertelement sequence is not folded into shuffle splat.
70 ; The length of scalable vector is unknown at compile-time. Therefore the following insertelements
71 ; may not form a valid splat.
72 define <vscale x 4 x float> @insertelement_sequene_may_not_be_splat(float %x) {
73 ; CHECK-LABEL: @insertelement_sequene_may_not_be_splat(
74 ; CHECK-NEXT: [[T0:%.*]] = insertelement <vscale x 4 x float> undef, float [[X:%.*]], i64 0
75 ; CHECK-NEXT: [[T1:%.*]] = insertelement <vscale x 4 x float> [[T0]], float [[X]], i64 1
76 ; CHECK-NEXT: [[T2:%.*]] = insertelement <vscale x 4 x float> [[T1]], float [[X]], i64 2
77 ; CHECK-NEXT: [[T3:%.*]] = insertelement <vscale x 4 x float> [[T2]], float [[X]], i64 3
78 ; CHECK-NEXT: ret <vscale x 4 x float> [[T3]]
79 ;
80 %t0 = insertelement <vscale x 4 x float> undef, float %x, i32 0
81 %t1 = insertelement <vscale x 4 x float> %t0, float %x, i32 1
82 %t2 = insertelement <vscale x 4 x float> %t1, float %x, i32 2
83 %t3 = insertelement <vscale x 4 x float> %t2, float %x, i32 3
84 ret <vscale x 4 x float> %t3
85 }
87 ; OSS-Fuzz #27416
88 ; https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=27416
89 define void @ossfuzz_27416(i32 %v, ptr %p) {
90 ; CHECK-LABEL: @ossfuzz_27416(
91 ; CHECK-NEXT: [[IN:%.*]] = insertelement <vscale x 4 x i32> undef, i32 [[V:%.*]], i64 0
92 ; CHECK-NEXT: [[SPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[IN]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
93 ; CHECK-NEXT: [[I1:%.*]] = insertelement <vscale x 4 x i32> [[SPLAT]], i32 undef, i64 128
94 ; CHECK-NEXT: store <vscale x 4 x i32> [[I1]], ptr [[P:%.*]], align 16
95 ; CHECK-NEXT: ret void
96 ;
97 %in = insertelement <vscale x 4 x i32> undef, i32 %v, i32 0
98 %splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
99 %I1 = insertelement <vscale x 4 x i32> %splat, i32 undef, i8 -128
100 store <vscale x 4 x i32> %I1, ptr %p, align 16
101 ret void
102 }