xref: /llvm-project/llvm/test/Transforms/VectorCombine/AArch64/scalarize-scalable.ll (revision ee4d6c8bf05d97238453d128770d1febb9a62216) 
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt -passes=vector-combine -S %s | FileCheck %s
3
4define <vscale x 4 x float> @scalarize_scalable(float %0, float %1, float %2, float %3) {
5; CHECK-LABEL: @scalarize_scalable(
6; CHECK-NEXT:  entry:
7; CHECK-NEXT:    [[DOTSCALAR:%.*]] = fadd fast float [[TMP1:%.*]], [[TMP0:%.*]]
8; CHECK-NEXT:    [[DOTSCALAR1:%.*]] = fadd fast float [[TMP2:%.*]], [[DOTSCALAR]]
9; CHECK-NEXT:    [[DOTSCALAR2:%.*]] = fadd fast float [[TMP3:%.*]], [[DOTSCALAR1]]
10; CHECK-NEXT:    [[TMP4:%.*]] = insertelement <vscale x 4 x float> poison, float [[DOTSCALAR2]], i64 0
11; CHECK-NEXT:    [[TMP5:%.*]] = shufflevector <vscale x 4 x float> [[TMP4]], <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
12; CHECK-NEXT:    ret <vscale x 4 x float> [[TMP5]]
13;
14entry:
15  %broadcast.splatinsert = insertelement <vscale x 4 x float> poison, float %0, i64 0
16  %broadcast.splatinsert28 = insertelement <vscale x 4 x float> poison, float %1, i64 0
17  %broadcast.splatinsert30 = insertelement <vscale x 4 x float> poison, float %2, i64 0
18  %broadcast.splatinsert32 = insertelement <vscale x 4 x float> poison, float %3, i64 0
19  %4 = fadd fast <vscale x 4 x float> %broadcast.splatinsert28, %broadcast.splatinsert
20  %5 = fadd fast <vscale x 4 x float> %broadcast.splatinsert30, %4
21  %6 = fadd fast <vscale x 4 x float> %broadcast.splatinsert32, %5
22  %7 = shufflevector <vscale x 4 x float> %6, <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
23  ret <vscale x 4 x float> %7
24}
25
26define <vscale x 4 x i32> @scalarize_scalable_udiv(i32 %x, i32 %y) {
27; CHECK-LABEL: @scalarize_scalable_udiv(
28; CHECK-NEXT:    [[R_SCALAR:%.*]] = udiv i32 [[X:%.*]], [[Y:%.*]]
29; CHECK-NEXT:    [[R:%.*]] = insertelement <vscale x 4 x i32> poison, i32 [[R_SCALAR]], i64 0
30; CHECK-NEXT:    ret <vscale x 4 x i32> [[R]]
31;
32  %splatx = insertelement <vscale x 4 x i32> poison, i32 %x, i64 0
33  %splaty = insertelement <vscale x 4 x i32> poison, i32 %y, i64 0
34  %r = udiv <vscale x 4 x i32> %splatx, %splaty
35  ret <vscale x 4 x i32> %r
36}
37
38define <vscale x 4 x i1> @scalarize_scalable_icmp(i32 %x, i32 %y) {
39; CHECK-LABEL: @scalarize_scalable_icmp(
40; CHECK-NEXT:    [[R_SCALAR:%.*]] = icmp sgt i32 [[X:%.*]], [[Y:%.*]]
41; CHECK-NEXT:    [[R:%.*]] = insertelement <vscale x 4 x i1> poison, i1 [[R_SCALAR]], i64 0
42; CHECK-NEXT:    ret <vscale x 4 x i1> [[R]]
43;
44  %splatx = insertelement <vscale x 4 x i32> poison, i32 %x, i64 0
45  %splaty = insertelement <vscale x 4 x i32> poison, i32 %y, i64 0
46  %r = icmp sgt <vscale x 4 x i32> %splatx, %splaty
47  ret <vscale x 4 x i1> %r
48}
49