103 lines
5.5 KiB
LLVM
103 lines
5.5 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
|
|
; RUN: opt -S -instcombine < %s | FileCheck %s
|
|
|
|
; This test checks that bitcast is moved after insertelement when both vector and scalar are
|
|
; bitcast from the same element type.
|
|
; inselt (bitcast VecSrc), (bitcast ScalarSrc), IdxOp
|
|
; --> bitcast (inselt VecSrc, ScalarSrc, IdxOp)
|
|
define <vscale x 4 x float> @insertelement_bitcast(<vscale x 4 x i32> %vec, i32 %x) {
|
|
; CHECK-LABEL: @insertelement_bitcast(
|
|
; CHECK-NEXT: [[TMP1:%.*]] = insertelement <vscale x 4 x i32> [[VEC:%.*]], i32 [[X:%.*]], i32 0
|
|
; CHECK-NEXT: [[R:%.*]] = bitcast <vscale x 4 x i32> [[TMP1]] to <vscale x 4 x float>
|
|
; CHECK-NEXT: ret <vscale x 4 x float> [[R]]
|
|
;
|
|
%x_cast = bitcast i32 %x to float
|
|
%vec_cast = bitcast <vscale x 4 x i32> %vec to <vscale x 4 x float>
|
|
%r = insertelement <vscale x 4 x float> %vec_cast, float %x_cast, i32 0
|
|
ret <vscale x 4 x float> %r
|
|
}
|
|
|
|
; This test checks that code-path "Try to form a shuffle from a chain of extract-insert ops" is
|
|
; not taken when both extract and insert are scalable type.
|
|
; For scalable type, the vector length needed to create shuffle mask is not a compile-time constant.
|
|
; Meanwhile, for scalable type shuffle mask only support splat and undef in the current code base.
|
|
; Otherwise we crash at:
|
|
; "Assertion `isValidOperands(V1, V2, Mask) && "Invalid shuffle vector instruction operands!"' failed."
|
|
define <vscale x 4 x i32> @insertelement_extractelement(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
|
|
; CHECK-LABEL: @insertelement_extractelement(
|
|
; CHECK-NEXT: [[T0:%.*]] = extractelement <vscale x 4 x i32> [[A:%.*]], i32 1
|
|
; CHECK-NEXT: [[T1:%.*]] = insertelement <vscale x 4 x i32> [[B:%.*]], i32 [[T0]], i32 0
|
|
; CHECK-NEXT: ret <vscale x 4 x i32> [[T1]]
|
|
;
|
|
%t0 = extractelement <vscale x 4 x i32> %a, i32 1
|
|
%t1 = insertelement <vscale x 4 x i32> %b, i32 %t0, i32 0
|
|
ret <vscale x 4 x i32> %t1
|
|
}
|
|
|
|
; This test checks that we are not attempting to create a shuffle from extract/insert chain,
|
|
; when extract is from a scalable type, and the insert vector is fixed-length.
|
|
define <4 x i32> @insertelement_extractelement_fixed_vec_extract_from_scalable(<vscale x 4 x i32> %a, <4 x i32> %b) {
|
|
; CHECK-LABEL: @insertelement_extractelement_fixed_vec_extract_from_scalable(
|
|
; CHECK-NEXT: [[T0:%.*]] = extractelement <vscale x 4 x i32> [[A:%.*]], i32 1
|
|
; CHECK-NEXT: [[T1:%.*]] = insertelement <4 x i32> [[B:%.*]], i32 [[T0]], i32 0
|
|
; CHECK-NEXT: ret <4 x i32> [[T1]]
|
|
;
|
|
%t0 = extractelement <vscale x 4 x i32> %a, i32 1
|
|
%t1 = insertelement <4 x i32> %b, i32 %t0, i32 0
|
|
ret <4 x i32> %t1
|
|
}
|
|
|
|
; This test checks that the optimization "foldConstantInsEltInfoShuffle" is not taken for scalable type.
|
|
; Particularly the fold:
|
|
; insertelt (insertelt X, C1, CIndex1), C, CIndex
|
|
; --> shufflevector X, CVec', Mask'
|
|
; For scalable type, the vector length needed to create shuffle mask is not a compile-time constant.
|
|
; Meanwhile, for scalable type shuffle mask only support splat and undef in the current code base.
|
|
; Otherwise we crash at:
|
|
; "Assertion `isValidOperands(V1, V2, Mask) && "Invalid shuffle vector instruction operands!"' failed."
|
|
define <vscale x 4 x i32> @insertelement_insertelement(<vscale x 4 x i32> %vec) {
|
|
; CHECK-LABEL: @insertelement_insertelement(
|
|
; CHECK-NEXT: [[T0:%.*]] = insertelement <vscale x 4 x i32> [[VEC:%.*]], i32 1, i32 1
|
|
; CHECK-NEXT: [[T1:%.*]] = insertelement <vscale x 4 x i32> [[T0]], i32 2, i32 2
|
|
; CHECK-NEXT: ret <vscale x 4 x i32> [[T1]]
|
|
;
|
|
%t0 = insertelement <vscale x 4 x i32> %vec, i32 1, i32 1
|
|
%t1 = insertelement <vscale x 4 x i32> %t0, i32 2, i32 2
|
|
ret <vscale x 4 x i32> %t1
|
|
}
|
|
|
|
; This test checks that the following insertelement sequence is not folded into shuffle splat.
|
|
; The length of scalable vector is unknown at compile-time. Therefore the following insertelements
|
|
; may not form a valid splat.
|
|
define <vscale x 4 x float> @insertelement_sequene_may_not_be_splat(float %x) {
|
|
; CHECK-LABEL: @insertelement_sequene_may_not_be_splat(
|
|
; CHECK-NEXT: [[T0:%.*]] = insertelement <vscale x 4 x float> undef, float [[X:%.*]], i32 0
|
|
; CHECK-NEXT: [[T1:%.*]] = insertelement <vscale x 4 x float> [[T0]], float [[X]], i32 1
|
|
; CHECK-NEXT: [[T2:%.*]] = insertelement <vscale x 4 x float> [[T1]], float [[X]], i32 2
|
|
; CHECK-NEXT: [[T3:%.*]] = insertelement <vscale x 4 x float> [[T2]], float [[X]], i32 3
|
|
; CHECK-NEXT: ret <vscale x 4 x float> [[T3]]
|
|
;
|
|
%t0 = insertelement <vscale x 4 x float> undef, float %x, i32 0
|
|
%t1 = insertelement <vscale x 4 x float> %t0, float %x, i32 1
|
|
%t2 = insertelement <vscale x 4 x float> %t1, float %x, i32 2
|
|
%t3 = insertelement <vscale x 4 x float> %t2, float %x, i32 3
|
|
ret <vscale x 4 x float> %t3
|
|
}
|
|
|
|
; OSS-Fuzz #27416
|
|
; https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=27416
|
|
define void @ossfuzz_27416(i32 %v) {
|
|
; CHECK-LABEL: @ossfuzz_27416(
|
|
; CHECK-NEXT: [[IN:%.*]] = insertelement <vscale x 4 x i32> undef, i32 [[V:%.*]], i32 0
|
|
; CHECK-NEXT: [[SPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[IN]], <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
|
|
; CHECK-NEXT: [[I1:%.*]] = insertelement <vscale x 4 x i32> [[SPLAT]], i32 undef, i8 -128
|
|
; CHECK-NEXT: store <vscale x 4 x i32> [[I1]], <vscale x 4 x i32>* undef, align 16
|
|
; CHECK-NEXT: ret void
|
|
;
|
|
%in = insertelement <vscale x 4 x i32> undef, i32 %v, i32 0
|
|
%splat = shufflevector <vscale x 4 x i32> %in, <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
|
|
%I1 = insertelement <vscale x 4 x i32> %splat, i32 undef, i8 -128
|
|
store <vscale x 4 x i32> %I1, <vscale x 4 x i32>* undef, align 16
|
|
ret void
|
|
}
|