xref: /llvm-project/llvm/test/CodeGen/AArch64/aarch64-combine-add-sub-mul.ll (revision db158c7c830807caeeb0691739c41f1d522029e9) 
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2; RUN: llc < %s -mtriple=aarch64-none-linux-gnu -mattr=+sve | FileCheck %s
3
4define <2 x i64> @test_mul_add_2x64(<2 x i64> %a, <2 x i64> %b, <2 x i64> %c) {
5; CHECK-LABEL: test_mul_add_2x64:
6; CHECK:       // %bb.0:
7; CHECK-NEXT:    ptrue p0.d, vl2
8; CHECK-NEXT:    // kill: def $q0 killed $q0 def $z0
9; CHECK-NEXT:    // kill: def $q2 killed $q2 def $z2
10; CHECK-NEXT:    // kill: def $q1 killed $q1 def $z1
11; CHECK-NEXT:    mla z0.d, p0/m, z1.d, z2.d
12; CHECK-NEXT:    // kill: def $q0 killed $q0 killed $z0
13; CHECK-NEXT:    ret
14  %mul = mul <2 x i64> %b, %c
15  %add = add <2 x i64> %a, %mul
16  ret <2 x i64> %add
17}
18
19define <1 x i64> @test_mul_add_1x64(<1 x i64> %a, <1 x i64> %b, <1 x i64> %c) {
20; CHECK-LABEL: test_mul_add_1x64:
21; CHECK:       // %bb.0:
22; CHECK-NEXT:    ptrue p0.d, vl1
23; CHECK-NEXT:    // kill: def $d0 killed $d0 def $z0
24; CHECK-NEXT:    // kill: def $d2 killed $d2 def $z2
25; CHECK-NEXT:    // kill: def $d1 killed $d1 def $z1
26; CHECK-NEXT:    mla z0.d, p0/m, z1.d, z2.d
27; CHECK-NEXT:    // kill: def $d0 killed $d0 killed $z0
28; CHECK-NEXT:    ret
29  %mul = mul <1 x i64> %b, %c
30  %add = add <1 x i64> %mul, %a
31  ret <1 x i64> %add
32}
33
34define <2 x i64> @test_mul_sub_2x64(<2 x i64> %a, <2 x i64> %b, <2 x i64> %c) {
35; CHECK-LABEL: test_mul_sub_2x64:
36; CHECK:       // %bb.0:
37; CHECK-NEXT:    ptrue p0.d, vl2
38; CHECK-NEXT:    // kill: def $q0 killed $q0 def $z0
39; CHECK-NEXT:    // kill: def $q2 killed $q2 def $z2
40; CHECK-NEXT:    // kill: def $q1 killed $q1 def $z1
41; CHECK-NEXT:    mls z0.d, p0/m, z1.d, z2.d
42; CHECK-NEXT:    // kill: def $q0 killed $q0 killed $z0
43; CHECK-NEXT:    ret
44  %mul = mul <2 x i64> %b, %c
45  %sub = sub <2 x i64> %a, %mul
46  ret <2 x i64> %sub
47}
48
49define <2 x i64> @test_mul_sub_2x64_2(<2 x i64> %a, <2 x i64> %b, <2 x i64> %c, <2 x i64> %d) {
50; CHECK-LABEL: test_mul_sub_2x64_2:
51; CHECK:       // %bb.0:
52; CHECK-NEXT:    ptrue p0.d, vl2
53; CHECK-NEXT:    // kill: def $q0 killed $q0 def $z0
54; CHECK-NEXT:    // kill: def $q1 killed $q1 def $z1
55; CHECK-NEXT:    // kill: def $q3 killed $q3 def $z3
56; CHECK-NEXT:    // kill: def $q2 killed $q2 def $z2
57; CHECK-NEXT:    sdiv z0.d, p0/m, z0.d, z1.d
58; CHECK-NEXT:    movprfx z1, z2
59; CHECK-NEXT:    mul z1.d, p0/m, z1.d, z3.d
60; CHECK-NEXT:    sub v0.2d, v1.2d, v0.2d
61; CHECK-NEXT:    ret
62  %div = sdiv <2 x i64> %a, %b
63  %mul = mul <2 x i64> %c, %d
64  %sub = sub <2 x i64> %mul, %div
65  ret <2 x i64> %sub
66}
67
68define <2 x i64> @test_mul_sub_2x64_3(<2 x i64> %a, <2 x i64> %b, <2 x i64> %c, <2 x i64> %d) {
69; CHECK-LABEL: test_mul_sub_2x64_3:
70; CHECK:       // %bb.0:
71; CHECK-NEXT:    ptrue p0.d, vl2
72; CHECK-NEXT:    // kill: def $q0 killed $q0 def $z0
73; CHECK-NEXT:    // kill: def $q1 killed $q1 def $z1
74; CHECK-NEXT:    // kill: def $q3 killed $q3 def $z3
75; CHECK-NEXT:    // kill: def $q2 killed $q2 def $z2
76; CHECK-NEXT:    sdiv z0.d, p0/m, z0.d, z1.d
77; CHECK-NEXT:    mls z0.d, p0/m, z2.d, z3.d
78; CHECK-NEXT:    // kill: def $q0 killed $q0 killed $z0
79; CHECK-NEXT:    ret
80  %div = sdiv <2 x i64> %a, %b
81  %mul = mul <2 x i64> %c, %d
82  %sub = sub <2 x i64> %div, %mul
83  ret <2 x i64> %sub
84}
85
86define <1 x i64> @test_mul_sub_1x64(<1 x i64> %a, <1 x i64> %b, <1 x i64> %c) {
87; CHECK-LABEL: test_mul_sub_1x64:
88; CHECK:       // %bb.0:
89; CHECK-NEXT:    ptrue p0.d, vl1
90; CHECK-NEXT:    // kill: def $d1 killed $d1 def $z1
91; CHECK-NEXT:    // kill: def $d2 killed $d2 def $z2
92; CHECK-NEXT:    mul z1.d, p0/m, z1.d, z2.d
93; CHECK-NEXT:    sub d0, d1, d0
94; CHECK-NEXT:    ret
95  %mul = mul <1 x i64> %b, %c
96  %sub = sub <1 x i64> %mul, %a
97  ret <1 x i64> %sub
98}
99