xref: /llvm-project/llvm/test/CodeGen/X86/avx512fp16-combine-shuffle-fma.ll (revision 0ddca87b794d92fc38114df537c87673770497ff)
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 3
2; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=avx2 | FileCheck %s --check-prefix=AVX2
3; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=f16c,fma | FileCheck %s --check-prefix=F16C
4; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=avx512vl | FileCheck %s --check-prefix=F16C
5; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=avx512fp16 | FileCheck %s --check-prefix=FP16
6
7define <2 x half> @foo(<2 x half> %0) "unsafe-fp-math"="true" nounwind {
8; AVX2-LABEL: foo:
9; AVX2:       # %bb.0:
10; AVX2-NEXT:    subq $40, %rsp
11; AVX2-NEXT:    vmovdqa %xmm0, {{[-0-9]+}}(%r{{[sb]}}p) # 16-byte Spill
12; AVX2-NEXT:    vpsrld $16, %xmm0, %xmm0
13; AVX2-NEXT:    callq __extendhfsf2@PLT
14; AVX2-NEXT:    vmulss {{\.?LCPI[0-9]+_[0-9]+}}(%rip), %xmm0, %xmm0
15; AVX2-NEXT:    callq __truncsfhf2@PLT
16; AVX2-NEXT:    vmovss %xmm0, {{[-0-9]+}}(%r{{[sb]}}p) # 4-byte Spill
17; AVX2-NEXT:    vmovaps {{[-0-9]+}}(%r{{[sb]}}p), %xmm0 # 16-byte Reload
18; AVX2-NEXT:    callq __extendhfsf2@PLT
19; AVX2-NEXT:    vmovss %xmm0, {{[-0-9]+}}(%r{{[sb]}}p) # 4-byte Spill
20; AVX2-NEXT:    vmovss {{[-0-9]+}}(%r{{[sb]}}p), %xmm0 # 4-byte Reload
21; AVX2-NEXT:    # xmm0 = mem[0],zero,zero,zero
22; AVX2-NEXT:    callq __extendhfsf2@PLT
23; AVX2-NEXT:    vsubss {{[-0-9]+}}(%r{{[sb]}}p), %xmm0, %xmm0 # 4-byte Folded Reload
24; AVX2-NEXT:    callq __truncsfhf2@PLT
25; AVX2-NEXT:    addq $40, %rsp
26; AVX2-NEXT:    retq
27;
28; F16C-LABEL: foo:
29; F16C:       # %bb.0:
30; F16C-NEXT:    vpsrld $16, %xmm0, %xmm1
31; F16C-NEXT:    vcvtph2ps %xmm1, %ymm1
32; F16C-NEXT:    vmulps {{\.?LCPI[0-9]+_[0-9]+}}(%rip), %ymm1, %ymm1
33; F16C-NEXT:    vcvtps2ph $4, %ymm1, %xmm1
34; F16C-NEXT:    vcvtph2ps %xmm0, %ymm0
35; F16C-NEXT:    vcvtph2ps %xmm1, %ymm1
36; F16C-NEXT:    vsubps %ymm0, %ymm1, %ymm2
37; F16C-NEXT:    vcvtps2ph $4, %ymm2, %xmm2
38; F16C-NEXT:    vaddps %ymm0, %ymm1, %ymm0
39; F16C-NEXT:    vcvtps2ph $4, %ymm0, %xmm0
40; F16C-NEXT:    vpblendw {{.*#+}} xmm0 = xmm2[0],xmm0[1],xmm2[2,3,4,5,6,7]
41; F16C-NEXT:    vzeroupper
42; F16C-NEXT:    retq
43;
44; FP16-LABEL: foo:
45; FP16:       # %bb.0:
46; FP16-NEXT:    vpsrld $16, %xmm0, %xmm1
47; FP16-NEXT:    vfmaddsub231ph {{\.?LCPI[0-9]+_[0-9]+}}(%rip), %xmm1, %xmm0
48; FP16-NEXT:    retq
49  %2 = shufflevector <2 x half> %0, <2 x half> undef, <2 x i32> <i32 1, i32 2>
50  %3 = fmul fast <2 x half> %2, <half 0xH3D3A, half 0xH3854>
51  %4 = fsub fast <2 x half> %3, %0
52  %5 = fadd fast <2 x half> %3, %0
53  %6 = shufflevector <2 x half> %4, <2 x half> %5, <2 x i32> <i32 0, i32 3>
54  %7 = fadd fast <2 x half> %6, zeroinitializer
55  %8 = shufflevector <2 x half> undef, <2 x half> %7, <2 x i32> <i32 0, i32 3>
56  %9 = fsub fast <2 x half> %8, zeroinitializer
57  ret <2 x half> %9
58}
59