; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt -S -passes=slp-vectorizer -mtriple=aarch64--linux-gnu < %s | FileCheck %s
target datalayout = "e-m:e-i32:64-i128:128-n32:64-S128"
declare void @foo(i64, i64, i64, i64)
define void @test1(<4 x i16> %a, <4 x i16> %b, ptr %p) {
; Make sure types of sub and its sources are not extended.
; CHECK-LABEL: @test1(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[Z0:%.*]] = zext <4 x i16> [[A:%.*]] to <4 x i32>
; CHECK-NEXT: [[Z1:%.*]] = zext <4 x i16> [[B:%.*]] to <4 x i32>
; CHECK-NEXT: [[SUB0:%.*]] = sub <4 x i32> [[Z0]], [[Z1]]
; CHECK-NEXT: [[E0:%.*]] = extractelement <4 x i32> [[SUB0]], i32 0
; CHECK-NEXT: [[S0:%.*]] = sext i32 [[E0]] to i64
; CHECK-NEXT: [[GEP0:%.*]] = getelementptr inbounds i64, ptr [[P:%.*]], i64 [[S0]]
; CHECK-NEXT: [[LOAD0:%.*]] = load i64, ptr [[GEP0]], align 4
; CHECK-NEXT: [[TMP0:%.*]] = shufflevector <4 x i32> [[SUB0]], <4 x i32> poison, <2 x i32> <i32 1, i32 2>
; CHECK-NEXT: [[TMP1:%.*]] = extractelement <2 x i32> [[TMP0]], i32 0
; CHECK-NEXT: [[TMP2:%.*]] = sext i32 [[TMP1]] to i64
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds i64, ptr [[P]], i64 [[TMP2]]
; CHECK-NEXT: [[LOAD1:%.*]] = load i64, ptr [[GEP1]], align 4
; CHECK-NEXT: [[TMP3:%.*]] = extractelement <2 x i32> [[TMP0]], i32 1
; CHECK-NEXT: [[TMP4:%.*]] = sext i32 [[TMP3]] to i64
; CHECK-NEXT: [[GEP2:%.*]] = getelementptr inbounds i64, ptr [[P]], i64 [[TMP4]]
; CHECK-NEXT: [[LOAD2:%.*]] = load i64, ptr [[GEP2]], align 4
; CHECK-NEXT: [[E3:%.*]] = extractelement <4 x i32> [[SUB0]], i32 3
; CHECK-NEXT: [[S3:%.*]] = sext i32 [[E3]] to i64
; CHECK-NEXT: [[GEP3:%.*]] = getelementptr inbounds i64, ptr [[P]], i64 [[S3]]
; CHECK-NEXT: [[LOAD3:%.*]] = load i64, ptr [[GEP3]], align 4
; CHECK-NEXT: call void @foo(i64 [[LOAD0]], i64 [[LOAD1]], i64 [[LOAD2]], i64 [[LOAD3]])
; CHECK-NEXT: ret void
;
entry:
%z0 = zext <4 x i16> %a to <4 x i32>
%z1 = zext <4 x i16> %b to <4 x i32>
%sub0 = sub <4 x i32> %z0, %z1
%e0 = extractelement <4 x i32> %sub0, i32 0
%s0 = sext i32 %e0 to i64
%gep0 = getelementptr inbounds i64, ptr %p, i64 %s0
%load0 = load i64, ptr %gep0
%e1 = extractelement <4 x i32> %sub0, i32 1
%s1 = sext i32 %e1 to i64
%gep1 = getelementptr inbounds i64, ptr %p, i64 %s1
%load1 = load i64, ptr %gep1
%e2 = extractelement <4 x i32> %sub0, i32 2
%s2 = sext i32 %e2 to i64
%gep2 = getelementptr inbounds i64, ptr %p, i64 %s2
%load2 = load i64, ptr %gep2
%e3 = extractelement <4 x i32> %sub0, i32 3
%s3 = sext i32 %e3 to i64
%gep3 = getelementptr inbounds i64, ptr %p, i64 %s3
%load3 = load i64, ptr %gep3
call void @foo(i64 %load0, i64 %load1, i64 %load2, i64 %load3)
ret void
}
define void @test2(<4 x i16> %a, <4 x i16> %b, i64 %c0, i64 %c1, i64 %c2, i64 %c3, ptr %p) {
; CHECK-LABEL: @test2(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[Z0:%.*]] = zext <4 x i16> [[A:%.*]] to <4 x i32>
; CHECK-NEXT: [[Z1:%.*]] = zext <4 x i16> [[B:%.*]] to <4 x i32>
; CHECK-NEXT: [[SUB0:%.*]] = sub <4 x i32> [[Z0]], [[Z1]]
; CHECK-NEXT: [[E0:%.*]] = extractelement <4 x i32> [[SUB0]], i32 0
; CHECK-NEXT: [[S0:%.*]] = sext i32 [[E0]] to i64
; CHECK-NEXT: [[A0:%.*]] = add i64 [[S0]], [[C0:%.*]]
; CHECK-NEXT: [[GEP0:%.*]] = getelementptr inbounds i64, ptr [[P:%.*]], i64 [[A0]]
; CHECK-NEXT: [[LOAD0:%.*]] = load i64, ptr [[GEP0]], align 4
; CHECK-NEXT: [[E1:%.*]] = extractelement <4 x i32> [[SUB0]], i32 1
; CHECK-NEXT: [[S1:%.*]] = sext i32 [[E1]] to i64
; CHECK-NEXT: [[A1:%.*]] = add i64 [[S1]], [[C1:%.*]]
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds i64, ptr [[P]], i64 [[A1]]
; CHECK-NEXT: [[LOAD1:%.*]] = load i64, ptr [[GEP1]], align 4
; CHECK-NEXT: [[E2:%.*]] = extractelement <4 x i32> [[SUB0]], i32 2
; CHECK-NEXT: [[S2:%.*]] = sext i32 [[E2]] to i64
; CHECK-NEXT: [[A2:%.*]] = add i64 [[S2]], [[C2:%.*]]
; CHECK-NEXT: [[GEP2:%.*]] = getelementptr inbounds i64, ptr [[P]], i64 [[A2]]
; CHECK-NEXT: [[LOAD2:%.*]] = load i64, ptr [[GEP2]], align 4
; CHECK-NEXT: [[E3:%.*]] = extractelement <4 x i32> [[SUB0]], i32 3
; CHECK-NEXT: [[S3:%.*]] = sext i32 [[E3]] to i64
; CHECK-NEXT: [[A3:%.*]] = add i64 [[S3]], [[C3:%.*]]
; CHECK-NEXT: [[GEP3:%.*]] = getelementptr inbounds i64, ptr [[P]], i64 [[A3]]
; CHECK-NEXT: [[LOAD3:%.*]] = load i64, ptr [[GEP3]], align 4
; CHECK-NEXT: call void @foo(i64 [[LOAD0]], i64 [[LOAD1]], i64 [[LOAD2]], i64 [[LOAD3]])
; CHECK-NEXT: ret void
;
entry:
%z0 = zext <4 x i16> %a to <4 x i32>
%z1 = zext <4 x i16> %b to <4 x i32>
%sub0 = sub <4 x i32> %z0, %z1
%e0 = extractelement <4 x i32> %sub0, i32 0
%s0 = sext i32 %e0 to i64
%a0 = add i64 %s0, %c0
%gep0 = getelementptr inbounds i64, ptr %p, i64 %a0
%load0 = load i64, ptr %gep0
%e1 = extractelement <4 x i32> %sub0, i32 1
%s1 = sext i32 %e1 to i64
%a1 = add i64 %s1, %c1
%gep1 = getelementptr inbounds i64, ptr %p, i64 %a1
%load1 = load i64, ptr %gep1
%e2 = extractelement <4 x i32> %sub0, i32 2
%s2 = sext i32 %e2 to i64
%a2 = add i64 %s2, %c2
%gep2 = getelementptr inbounds i64, ptr %p, i64 %a2
%load2 = load i64, ptr %gep2
%e3 = extractelement <4 x i32> %sub0, i32 3
%s3 = sext i32 %e3 to i64
%a3 = add i64 %s3, %c3
%gep3 = getelementptr inbounds i64, ptr %p, i64 %a3
%load3 = load i64, ptr %gep3
call void @foo(i64 %load0, i64 %load1, i64 %load2, i64 %load3)
ret void
}