xref: /llvm-project/llvm/test/CodeGen/X86/prefer-avx256-popcnt.ll (revision f6ff2cc7e0ae4fd9b14583a998ddeada256a954f) 
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+avx512vl,+avx512vpopcntdq,+prefer-256-bit | FileCheck %s --check-prefix=CHECK --check-prefix=AVX256
3; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+avx512vl,+avx512vpopcntdq,-prefer-256-bit | FileCheck %s --check-prefix=CHECK --check-prefix=AVX512 --check-prefix=AVX512VL
4; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+avx512f,+avx512vpopcntdq,+prefer-256-bit | FileCheck %s --check-prefix=CHECK --check-prefix=AVX512 --check-prefix=AVX512F
5; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+avx512f,+avx512vpopcntdq,-prefer-256-bit | FileCheck %s --check-prefix=CHECK --check-prefix=AVX512 --check-prefix=AVX512F
6
7define <8 x i16> @testv8i16(<8 x i16> %in) {
8; AVX256-LABEL: testv8i16:
9; AVX256:       # %bb.0:
10; AVX256-NEXT:    vpmovzxwd {{.*#+}} 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
11; AVX256-NEXT:    vpopcntd %ymm0, %ymm0
12; AVX256-NEXT:    vpmovdw %ymm0, %xmm0
13; AVX256-NEXT:    vzeroupper
14; AVX256-NEXT:    retq
15;
16; AVX512VL-LABEL: testv8i16:
17; AVX512VL:       # %bb.0:
18; AVX512VL-NEXT:    vpmovzxwd {{.*#+}} 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
19; AVX512VL-NEXT:    vpopcntd %ymm0, %ymm0
20; AVX512VL-NEXT:    vpmovdw %ymm0, %xmm0
21; AVX512VL-NEXT:    vzeroupper
22; AVX512VL-NEXT:    retq
23;
24; AVX512F-LABEL: testv8i16:
25; AVX512F:       # %bb.0:
26; AVX512F-NEXT:    vpmovzxwd {{.*#+}} 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
27; AVX512F-NEXT:    vpopcntd %zmm0, %zmm0
28; AVX512F-NEXT:    vpmovdw %zmm0, %ymm0
29; AVX512F-NEXT:    # kill: def $xmm0 killed $xmm0 killed $ymm0
30; AVX512F-NEXT:    vzeroupper
31; AVX512F-NEXT:    retq
32  %out = call <8 x i16> @llvm.ctpop.v8i16(<8 x i16> %in)
33  ret <8 x i16> %out
34}
35
36define <16 x i8> @testv16i8(<16 x i8> %in) {
37; AVX256-LABEL: testv16i8:
38; AVX256:       # %bb.0:
39; AVX256-NEXT:    vpbroadcastd {{.*#+}} xmm1 = [15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15]
40; AVX256-NEXT:    vpand %xmm1, %xmm0, %xmm2
41; AVX256-NEXT:    vmovdqa {{.*#+}} xmm3 = [0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4]
42; AVX256-NEXT:    vpshufb %xmm2, %xmm3, %xmm2
43; AVX256-NEXT:    vpsrlw $4, %xmm0, %xmm0
44; AVX256-NEXT:    vpand %xmm1, %xmm0, %xmm0
45; AVX256-NEXT:    vpshufb %xmm0, %xmm3, %xmm0
46; AVX256-NEXT:    vpaddb %xmm2, %xmm0, %xmm0
47; AVX256-NEXT:    retq
48;
49; AVX512-LABEL: testv16i8:
50; AVX512:       # %bb.0:
51; AVX512-NEXT:    vpmovzxbd {{.*#+}} zmm0 = xmm0[0],zero,zero,zero,xmm0[1],zero,zero,zero,xmm0[2],zero,zero,zero,xmm0[3],zero,zero,zero,xmm0[4],zero,zero,zero,xmm0[5],zero,zero,zero,xmm0[6],zero,zero,zero,xmm0[7],zero,zero,zero,xmm0[8],zero,zero,zero,xmm0[9],zero,zero,zero,xmm0[10],zero,zero,zero,xmm0[11],zero,zero,zero,xmm0[12],zero,zero,zero,xmm0[13],zero,zero,zero,xmm0[14],zero,zero,zero,xmm0[15],zero,zero,zero
52; AVX512-NEXT:    vpopcntd %zmm0, %zmm0
53; AVX512-NEXT:    vpmovdb %zmm0, %xmm0
54; AVX512-NEXT:    vzeroupper
55; AVX512-NEXT:    retq
56  %out = call <16 x i8> @llvm.ctpop.v16i8(<16 x i8> %in)
57  ret <16 x i8> %out
58}
59
60define <16 x i16> @testv16i16(<16 x i16> %in) {
61; AVX256-LABEL: testv16i16:
62; AVX256:       # %bb.0:
63; AVX256-NEXT:    vpbroadcastd {{.*#+}} ymm1 = [15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15]
64; AVX256-NEXT:    vpand %ymm1, %ymm0, %ymm2
65; AVX256-NEXT:    vbroadcasti128 {{.*#+}} ymm3 = [0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4,0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4]
66; AVX256-NEXT:    # ymm3 = mem[0,1,0,1]
67; AVX256-NEXT:    vpshufb %ymm2, %ymm3, %ymm2
68; AVX256-NEXT:    vpsrlw $4, %ymm0, %ymm0
69; AVX256-NEXT:    vpand %ymm1, %ymm0, %ymm0
70; AVX256-NEXT:    vpshufb %ymm0, %ymm3, %ymm0
71; AVX256-NEXT:    vpaddb %ymm2, %ymm0, %ymm0
72; AVX256-NEXT:    vpsllw $8, %ymm0, %ymm1
73; AVX256-NEXT:    vpaddb %ymm0, %ymm1, %ymm0
74; AVX256-NEXT:    vpsrlw $8, %ymm0, %ymm0
75; AVX256-NEXT:    retq
76;
77; AVX512-LABEL: testv16i16:
78; AVX512:       # %bb.0:
79; 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
80; AVX512-NEXT:    vpopcntd %zmm0, %zmm0
81; AVX512-NEXT:    vpmovdw %zmm0, %ymm0
82; AVX512-NEXT:    retq
83  %out = call <16 x i16> @llvm.ctpop.v16i16(<16 x i16> %in)
84  ret <16 x i16> %out
85}
86
87define <32 x i8> @testv32i8(<32 x i8> %in) {
88; AVX256-LABEL: testv32i8:
89; AVX256:       # %bb.0:
90; AVX256-NEXT:    vpbroadcastd {{.*#+}} ymm1 = [15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15]
91; AVX256-NEXT:    vpand %ymm1, %ymm0, %ymm2
92; AVX256-NEXT:    vbroadcasti128 {{.*#+}} ymm3 = [0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4,0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4]
93; AVX256-NEXT:    # ymm3 = mem[0,1,0,1]
94; AVX256-NEXT:    vpshufb %ymm2, %ymm3, %ymm2
95; AVX256-NEXT:    vpsrlw $4, %ymm0, %ymm0
96; AVX256-NEXT:    vpand %ymm1, %ymm0, %ymm0
97; AVX256-NEXT:    vpshufb %ymm0, %ymm3, %ymm0
98; AVX256-NEXT:    vpaddb %ymm2, %ymm0, %ymm0
99; AVX256-NEXT:    retq
100;
101; AVX512VL-LABEL: testv32i8:
102; AVX512VL:       # %bb.0:
103; AVX512VL-NEXT:    vpbroadcastd {{.*#+}} ymm1 = [15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15]
104; AVX512VL-NEXT:    vpand %ymm1, %ymm0, %ymm2
105; AVX512VL-NEXT:    vbroadcasti128 {{.*#+}} ymm3 = [0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4,0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4]
106; AVX512VL-NEXT:    # ymm3 = mem[0,1,0,1]
107; AVX512VL-NEXT:    vpshufb %ymm2, %ymm3, %ymm2
108; AVX512VL-NEXT:    vpsrlw $4, %ymm0, %ymm0
109; AVX512VL-NEXT:    vpand %ymm1, %ymm0, %ymm0
110; AVX512VL-NEXT:    vpshufb %ymm0, %ymm3, %ymm0
111; AVX512VL-NEXT:    vpaddb %ymm2, %ymm0, %ymm0
112; AVX512VL-NEXT:    retq
113;
114; AVX512F-LABEL: testv32i8:
115; AVX512F:       # %bb.0:
116; AVX512F-NEXT:    vpbroadcastb {{.*#+}} ymm1 = [15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15]
117; AVX512F-NEXT:    vpand %ymm1, %ymm0, %ymm2
118; AVX512F-NEXT:    vbroadcasti128 {{.*#+}} ymm3 = [0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4,0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4]
119; AVX512F-NEXT:    # ymm3 = mem[0,1,0,1]
120; AVX512F-NEXT:    vpshufb %ymm2, %ymm3, %ymm2
121; AVX512F-NEXT:    vpsrlw $4, %ymm0, %ymm0
122; AVX512F-NEXT:    vpand %ymm1, %ymm0, %ymm0
123; AVX512F-NEXT:    vpshufb %ymm0, %ymm3, %ymm0
124; AVX512F-NEXT:    vpaddb %ymm2, %ymm0, %ymm0
125; AVX512F-NEXT:    retq
126  %out = call <32 x i8> @llvm.ctpop.v32i8(<32 x i8> %in)
127  ret <32 x i8> %out
128}
129
130declare <8 x i16> @llvm.ctpop.v8i16(<8 x i16>)
131declare <16 x i8> @llvm.ctpop.v16i8(<16 x i8>)
132declare <16 x i16> @llvm.ctpop.v16i16(<16 x i16>)
133declare <32 x i8> @llvm.ctpop.v32i8(<32 x i8>)
134;; NOTE: These prefixes are unused and the list is autogenerated. Do not add tests below this line:
135; CHECK: {{.*}}
136