xref: /llvm-project/llvm/test/Analysis/CostModel/AArch64/sve-widening-instruction.ll (revision 73668cceaee51193bf954c3f36ef18e65b1cbaf9)
1; NOTE: Assertions have been autogenerated by utils/update_analyze_test_checks.py
2; RUN: opt  -passes="print<cost-model>" 2>&1 -disable-output -mtriple=aarch64--linux-gnu -mattr=+sve < %s | FileCheck %s
3
4; Checks if widening instructions works for SVE
5
6target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
7
8define <vscale x 4 x i32> @widening_nxv4i16(<vscale x 4 x i16> %in1, <vscale x 4 x i16> %in2) {
9; CHECK-LABEL: 'widening_nxv4i16'
10; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %in1.ext = zext <vscale x 4 x i16> %in2 to <vscale x 4 x i32>
11; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %in2.ext = zext <vscale x 4 x i16> %in2 to <vscale x 4 x i32>
12; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %in.add = add <vscale x 4 x i32> %in1.ext, %in2.ext
13; CHECK-NEXT:  Cost Model: Found an estimated cost of 0 for instruction: ret <vscale x 4 x i32> %in.add
14;
15  %in1.ext = zext <vscale x 4 x i16> %in2 to <vscale x 4 x i32>
16  %in2.ext = zext <vscale x 4 x i16> %in2 to <vscale x 4 x i32>
17  %in.add = add <vscale x 4 x i32> %in1.ext, %in2.ext
18  ret <vscale x 4 x i32> %in.add
19}
20
21define <vscale x 8 x i32> @widening_nxv8i16(<vscale x 8 x i16> %in1, <vscale x 8 x i16> %in2) {
22; CHECK-LABEL: 'widening_nxv8i16'
23; CHECK-NEXT:  Cost Model: Found an estimated cost of 2 for instruction: %in1.ext = zext <vscale x 8 x i16> %in2 to <vscale x 8 x i32>
24; CHECK-NEXT:  Cost Model: Found an estimated cost of 2 for instruction: %in2.ext = zext <vscale x 8 x i16> %in2 to <vscale x 8 x i32>
25; CHECK-NEXT:  Cost Model: Found an estimated cost of 2 for instruction: %in.add = add <vscale x 8 x i32> %in1.ext, %in2.ext
26; CHECK-NEXT:  Cost Model: Found an estimated cost of 0 for instruction: ret <vscale x 8 x i32> %in.add
27;
28  %in1.ext = zext <vscale x 8 x i16> %in2 to <vscale x 8 x i32>
29  %in2.ext = zext <vscale x 8 x i16> %in2 to <vscale x 8 x i32>
30  %in.add = add <vscale x 8 x i32> %in1.ext, %in2.ext
31  ret <vscale x 8 x i32> %in.add
32}
33
34define <8 x i32> @widening_v8i16_svevl2(<8 x i16> %in1, <8 x i16> %in2) vscale_range(2,16) {
35; CHECK-LABEL: 'widening_v8i16_svevl2'
36; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %in1.ext = zext <8 x i16> %in2 to <8 x i32>
37; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %in2.ext = zext <8 x i16> %in2 to <8 x i32>
38; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %in.add = add <8 x i32> %in1.ext, %in2.ext
39; CHECK-NEXT:  Cost Model: Found an estimated cost of 0 for instruction: ret <8 x i32> %in.add
40;
41  %in1.ext = zext <8 x i16> %in2 to <8 x i32>
42  %in2.ext = zext <8 x i16> %in2 to <8 x i32>
43  %in.add = add <8 x i32> %in1.ext, %in2.ext
44  ret <8 x i32> %in.add
45}
46
47