| blob | e3a608abfda4332c18214e9c74e646ff8acf74af |
1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2 ; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+avx512vl,+prefer-256-bit | FileCheck %s --check-prefix=AVX256
3 ; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+avx512vl,-prefer-256-bit | FileCheck %s --check-prefix=AVX512
5 define <16 x i1> @smulo_v16i8(<16 x i8> %a0, <16 x i8> %a1, ptr %p2) nounwind {
6 ; AVX256-LABEL: smulo_v16i8:
7 ; AVX256: # %bb.0:
8 ; AVX256-NEXT: vpmovsxbw %xmm1, %ymm1
9 ; AVX256-NEXT: vpmovsxbw %xmm0, %ymm0
10 ; AVX256-NEXT: vpmullw %ymm1, %ymm0, %ymm0
11 ; AVX256-NEXT: vpsrlw $8, %ymm0, %ymm1
12 ; AVX256-NEXT: vextracti128 $1, %ymm1, %xmm2
13 ; AVX256-NEXT: vpackuswb %xmm2, %xmm1, %xmm1
14 ; AVX256-NEXT: vpandd {{\.?LCPI[0-9]+_[0-9]+}}(%rip){1to8}, %ymm0, %ymm0
15 ; AVX256-NEXT: vextracti128 $1, %ymm0, %xmm2
16 ; AVX256-NEXT: vpackuswb %xmm2, %xmm0, %xmm0
17 ; AVX256-NEXT: vpxor %xmm2, %xmm2, %xmm2
18 ; AVX256-NEXT: vpcmpgtb %xmm0, %xmm2, %xmm2
19 ; AVX256-NEXT: vpcmpeqb %xmm1, %xmm2, %xmm1
20 ; AVX256-NEXT: vpternlogq $15, %xmm1, %xmm1, %xmm1
21 ; AVX256-NEXT: vpshufd {{.*#+}} xmm2 = xmm1[2,3,2,3]
22 ; AVX256-NEXT: vpmovsxbd %xmm2, %ymm2
23 ; AVX256-NEXT: vptestmd %ymm2, %ymm2, %k1
24 ; AVX256-NEXT: vpmovsxbd %xmm1, %ymm1
25 ; AVX256-NEXT: vptestmd %ymm1, %ymm1, %k2
26 ; AVX256-NEXT: vmovdqa %xmm0, (%rdi)
27 ; AVX256-NEXT: vpcmpeqd %ymm0, %ymm0, %ymm0
28 ; AVX256-NEXT: vmovdqa32 %ymm0, %ymm1 {%k2} {z}
29 ; AVX256-NEXT: vpmovdw %ymm1, %xmm1
30 ; AVX256-NEXT: vmovdqa32 %ymm0, %ymm0 {%k1} {z}
31 ; AVX256-NEXT: vpmovdw %ymm0, %xmm0
32 ; AVX256-NEXT: vpacksswb %xmm0, %xmm1, %xmm0
33 ; AVX256-NEXT: vzeroupper
34 ; AVX256-NEXT: retq
35 ;
36 ; AVX512-LABEL: smulo_v16i8:
37 ; AVX512: # %bb.0:
38 ; AVX512-NEXT: vpmovsxbw %xmm1, %ymm1
39 ; AVX512-NEXT: vpmovsxbw %xmm0, %ymm0
40 ; AVX512-NEXT: vpmullw %ymm1, %ymm0, %ymm0
41 ; AVX512-NEXT: vpsraw $8, %ymm0, %ymm1
42 ; AVX512-NEXT: vpmovsxwd %ymm1, %zmm1
43 ; AVX512-NEXT: vpsllw $8, %ymm0, %ymm2
44 ; AVX512-NEXT: vpsraw $15, %ymm2, %ymm2
45 ; AVX512-NEXT: vpmovsxwd %ymm2, %zmm2
46 ; AVX512-NEXT: vpcmpneqd %zmm1, %zmm2, %k1
47 ; AVX512-NEXT: vpmovzxwd {{.*#+}} zmm0 = ymm0[0],zero,ymm0[1],zero,ymm0[2],zero,ymm0[3],zero,ymm0[4],zero,ymm0[5],zero,ymm0[6],zero,ymm0[7],zero,ymm0[8],zero,ymm0[9],zero,ymm0[10],zero,ymm0[11],zero,ymm0[12],zero,ymm0[13],zero,ymm0[14],zero,ymm0[15],zero
48 ; AVX512-NEXT: vpmovdb %zmm0, (%rdi)
49 ; AVX512-NEXT: vpternlogd $255, %zmm0, %zmm0, %zmm0 {%k1} {z}
50 ; AVX512-NEXT: vpmovdb %zmm0, %xmm0
51 ; AVX512-NEXT: vzeroupper
52 ; AVX512-NEXT: retq
53 %t = call {<16 x i8>, <16 x i1>} @llvm.smul.with.overflow.v16i8(<16 x i8> %a0, <16 x i8> %a1)
54 %val = extractvalue {<16 x i8>, <16 x i1>} %t, 0
55 %obit = extractvalue {<16 x i8>, <16 x i1>} %t, 1
56 store <16 x i8> %val, ptr %p2
57 ret <16 x i1> %obit
58 }
59 declare {<16 x i8>, <16 x i1>} @llvm.smul.with.overflow.v16i8(<16 x i8>, <16 x i8>)
61 define <16 x i1> @umulo_v16i8(<16 x i8> %a0, <16 x i8> %a1, ptr %p2) nounwind {
62 ; AVX256-LABEL: umulo_v16i8:
63 ; AVX256: # %bb.0:
64 ; AVX256-NEXT: vpmovzxbw {{.*#+}} ymm1 = xmm1[0],zero,xmm1[1],zero,xmm1[2],zero,xmm1[3],zero,xmm1[4],zero,xmm1[5],zero,xmm1[6],zero,xmm1[7],zero,xmm1[8],zero,xmm1[9],zero,xmm1[10],zero,xmm1[11],zero,xmm1[12],zero,xmm1[13],zero,xmm1[14],zero,xmm1[15],zero
65 ; AVX256-NEXT: vpmovzxbw {{.*#+}} ymm0 = xmm0[0],zero,xmm0[1],zero,xmm0[2],zero,xmm0[3],zero,xmm0[4],zero,xmm0[5],zero,xmm0[6],zero,xmm0[7],zero,xmm0[8],zero,xmm0[9],zero,xmm0[10],zero,xmm0[11],zero,xmm0[12],zero,xmm0[13],zero,xmm0[14],zero,xmm0[15],zero
66 ; AVX256-NEXT: vpmullw %ymm1, %ymm0, %ymm0
67 ; AVX256-NEXT: vpsrlw $8, %ymm0, %ymm1
68 ; AVX256-NEXT: vextracti128 $1, %ymm1, %xmm2
69 ; AVX256-NEXT: vpackuswb %xmm2, %xmm1, %xmm1
70 ; AVX256-NEXT: vpxor %xmm2, %xmm2, %xmm2
71 ; AVX256-NEXT: vpcmpeqb %xmm2, %xmm1, %xmm1
72 ; AVX256-NEXT: vpternlogq $15, %xmm1, %xmm1, %xmm1
73 ; AVX256-NEXT: vpshufd {{.*#+}} xmm2 = xmm1[2,3,2,3]
74 ; AVX256-NEXT: vpmovsxbd %xmm2, %ymm2
75 ; AVX256-NEXT: vptestmd %ymm2, %ymm2, %k1
76 ; AVX256-NEXT: vpmovsxbd %xmm1, %ymm1
77 ; AVX256-NEXT: vptestmd %ymm1, %ymm1, %k2
78 ; AVX256-NEXT: vpandd {{\.?LCPI[0-9]+_[0-9]+}}(%rip){1to8}, %ymm0, %ymm0
79 ; AVX256-NEXT: vextracti128 $1, %ymm0, %xmm1
80 ; AVX256-NEXT: vpackuswb %xmm1, %xmm0, %xmm0
81 ; AVX256-NEXT: vmovdqa %xmm0, (%rdi)
82 ; AVX256-NEXT: vpcmpeqd %ymm0, %ymm0, %ymm0
83 ; AVX256-NEXT: vmovdqa32 %ymm0, %ymm1 {%k2} {z}
84 ; AVX256-NEXT: vpmovdw %ymm1, %xmm1
85 ; AVX256-NEXT: vmovdqa32 %ymm0, %ymm0 {%k1} {z}
86 ; AVX256-NEXT: vpmovdw %ymm0, %xmm0
87 ; AVX256-NEXT: vpacksswb %xmm0, %xmm1, %xmm0
88 ; AVX256-NEXT: vzeroupper
89 ; AVX256-NEXT: retq
90 ;
91 ; AVX512-LABEL: umulo_v16i8:
92 ; AVX512: # %bb.0:
93 ; AVX512-NEXT: vpmovzxbw {{.*#+}} ymm1 = xmm1[0],zero,xmm1[1],zero,xmm1[2],zero,xmm1[3],zero,xmm1[4],zero,xmm1[5],zero,xmm1[6],zero,xmm1[7],zero,xmm1[8],zero,xmm1[9],zero,xmm1[10],zero,xmm1[11],zero,xmm1[12],zero,xmm1[13],zero,xmm1[14],zero,xmm1[15],zero
94 ; AVX512-NEXT: vpmovzxbw {{.*#+}} ymm0 = xmm0[0],zero,xmm0[1],zero,xmm0[2],zero,xmm0[3],zero,xmm0[4],zero,xmm0[5],zero,xmm0[6],zero,xmm0[7],zero,xmm0[8],zero,xmm0[9],zero,xmm0[10],zero,xmm0[11],zero,xmm0[12],zero,xmm0[13],zero,xmm0[14],zero,xmm0[15],zero
95 ; AVX512-NEXT: vpmullw %ymm1, %ymm0, %ymm0
96 ; AVX512-NEXT: vpsrlw $8, %ymm0, %ymm1
97 ; AVX512-NEXT: vpmovzxwd {{.*#+}} zmm1 = ymm1[0],zero,ymm1[1],zero,ymm1[2],zero,ymm1[3],zero,ymm1[4],zero,ymm1[5],zero,ymm1[6],zero,ymm1[7],zero,ymm1[8],zero,ymm1[9],zero,ymm1[10],zero,ymm1[11],zero,ymm1[12],zero,ymm1[13],zero,ymm1[14],zero,ymm1[15],zero
98 ; AVX512-NEXT: vptestmd %zmm1, %zmm1, %k1
99 ; AVX512-NEXT: vpmovzxwd {{.*#+}} zmm0 = ymm0[0],zero,ymm0[1],zero,ymm0[2],zero,ymm0[3],zero,ymm0[4],zero,ymm0[5],zero,ymm0[6],zero,ymm0[7],zero,ymm0[8],zero,ymm0[9],zero,ymm0[10],zero,ymm0[11],zero,ymm0[12],zero,ymm0[13],zero,ymm0[14],zero,ymm0[15],zero
100 ; AVX512-NEXT: vpmovdb %zmm0, (%rdi)
101 ; AVX512-NEXT: vpternlogd $255, %zmm0, %zmm0, %zmm0 {%k1} {z}
102 ; AVX512-NEXT: vpmovdb %zmm0, %xmm0
103 ; AVX512-NEXT: vzeroupper
104 ; AVX512-NEXT: retq
105 %t = call {<16 x i8>, <16 x i1>} @llvm.umul.with.overflow.v16i8(<16 x i8> %a0, <16 x i8> %a1)
106 %val = extractvalue {<16 x i8>, <16 x i1>} %t, 0
107 %obit = extractvalue {<16 x i8>, <16 x i1>} %t, 1
108 store <16 x i8> %val, ptr %p2
109 ret <16 x i1> %obit
110 }
111 declare {<16 x i8>, <16 x i1>} @llvm.umul.with.overflow.v16i8(<16 x i8>, <16 x i8>)