Anton Afanasyev 954ea0f044 [SLP] Simplify indices processing for insertelements
Get rid of non-constant and undef indices of insertelements
at `buildTree()` stage. Fix bugs.

Differential Revision: https://reviews.llvm.org/D119623
2022-02-14 14:50:44 +03:00

68 lines
2.6 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt -S -slp-vectorizer < %s | FileCheck %s
; These all crashing before patch
define <2 x i8> @negative_index(<2 x i8> %aux_vec, i8* %in) {
; CHECK-LABEL: @negative_index(
; CHECK-NEXT: [[IN0:%.*]] = load i8, i8* [[IN:%.*]], align 4
; CHECK-NEXT: [[G1:%.*]] = getelementptr inbounds i8, i8* [[IN]], i64 1
; CHECK-NEXT: [[IN1:%.*]] = load i8, i8* [[G1]], align 4
; CHECK-NEXT: [[V0:%.*]] = insertelement <2 x i8> [[AUX_VEC:%.*]], i8 [[IN0]], i8 -1
; CHECK-NEXT: [[V1:%.*]] = insertelement <2 x i8> [[V0]], i8 [[IN1]], i64 1
; CHECK-NEXT: [[OUT:%.*]] = add <2 x i8> [[V1]], [[V1]]
; CHECK-NEXT: ret <2 x i8> [[OUT]]
;
%in0 = load i8, i8* %in, align 4
%g1 = getelementptr inbounds i8, i8* %in, i64 1
%in1 = load i8, i8* %g1, align 4
%v0 = insertelement <2 x i8> %aux_vec, i8 %in0, i8 -1
%v1 = insertelement <2 x i8> %v0, i8 %in1, i64 1
%out = add <2 x i8> %v1, %v1
ret <2 x i8> %out
}
define <2 x i8> @exceed_index(<2 x i8> %aux_vec, i8* %in) {
; CHECK-LABEL: @exceed_index(
; CHECK-NEXT: [[IN0:%.*]] = load i8, i8* [[IN:%.*]], align 4
; CHECK-NEXT: [[G1:%.*]] = getelementptr inbounds i8, i8* [[IN]], i64 1
; CHECK-NEXT: [[IN1:%.*]] = load i8, i8* [[G1]], align 4
; CHECK-NEXT: [[V0:%.*]] = insertelement <2 x i8> [[AUX_VEC:%.*]], i8 [[IN0]], i8 2
; CHECK-NEXT: [[V1:%.*]] = insertelement <2 x i8> [[V0]], i8 [[IN1]], i64 1
; CHECK-NEXT: [[OUT:%.*]] = add <2 x i8> [[V1]], [[V1]]
; CHECK-NEXT: ret <2 x i8> [[OUT]]
;
%in0 = load i8, i8* %in, align 4
%g1 = getelementptr inbounds i8, i8* %in, i64 1
%in1 = load i8, i8* %g1, align 4
%v0 = insertelement <2 x i8> %aux_vec, i8 %in0, i8 2
%v1 = insertelement <2 x i8> %v0, i8 %in1, i64 1
%out = add <2 x i8> %v1, %v1
ret <2 x i8> %out
}
define <2 x i8> @poison_index(<2 x i8> %aux_vec, i8* %in) {
; CHECK-LABEL: @poison_index(
; CHECK-NEXT: [[IN0:%.*]] = load i8, i8* [[IN:%.*]], align 4
; CHECK-NEXT: [[G1:%.*]] = getelementptr inbounds i8, i8* [[IN]], i64 1
; CHECK-NEXT: [[IN1:%.*]] = load i8, i8* [[G1]], align 4
; CHECK-NEXT: [[V0:%.*]] = insertelement <2 x i8> [[AUX_VEC:%.*]], i8 [[IN0]], i8 poison
; CHECK-NEXT: [[V1:%.*]] = insertelement <2 x i8> [[V0]], i8 [[IN1]], i64 1
; CHECK-NEXT: [[OUT:%.*]] = add <2 x i8> [[V1]], [[V1]]
; CHECK-NEXT: ret <2 x i8> [[OUT]]
;
%in0 = load i8, i8* %in, align 4
%g1 = getelementptr inbounds i8, i8* %in, i64 1
%in1 = load i8, i8* %g1, align 4
%v0 = insertelement <2 x i8> %aux_vec, i8 %in0, i8 poison
%v1 = insertelement <2 x i8> %v0, i8 %in1, i64 1
%out = add <2 x i8> %v1, %v1
ret <2 x i8> %out
}