]>
git.proxmox.com Git - mirror_ubuntu-jammy-kernel.git/blob - lib/raid6/avx2.c
1 /* -*- linux-c -*- ------------------------------------------------------- *
3 * Copyright (C) 2012 Intel Corporation
4 * Author: Yuanhan Liu <yuanhan.liu@linux.intel.com>
6 * Based on sse2.c: Copyright 2002 H. Peter Anvin - All Rights Reserved
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation, Inc., 53 Temple Place Ste 330,
12 * Boston MA 02111-1307, USA; either version 2 of the License, or
13 * (at your option) any later version; incorporated herein by reference.
15 * ----------------------------------------------------------------------- */
18 * AVX2 implementation of RAID-6 syndrome functions
24 #include <linux/raid/pq.h>
27 static const struct raid6_avx2_constants
{
29 } raid6_avx2_constants
__aligned(32) = {
30 { 0x1d1d1d1d1d1d1d1dULL
, 0x1d1d1d1d1d1d1d1dULL
,
31 0x1d1d1d1d1d1d1d1dULL
, 0x1d1d1d1d1d1d1d1dULL
,},
34 static int raid6_have_avx2(void)
36 return boot_cpu_has(X86_FEATURE_AVX2
) && boot_cpu_has(X86_FEATURE_AVX
);
40 * Plain AVX2 implementation
42 static void raid6_avx21_gen_syndrome(int disks
, size_t bytes
, void **ptrs
)
44 u8
**dptr
= (u8
**)ptrs
;
48 z0
= disks
- 3; /* Highest data disk */
49 p
= dptr
[z0
+1]; /* XOR parity */
50 q
= dptr
[z0
+2]; /* RS syndrome */
54 asm volatile("vmovdqa %0,%%ymm0" : : "m" (raid6_avx2_constants
.x1d
[0]));
55 asm volatile("vpxor %ymm3,%ymm3,%ymm3"); /* Zero temp */
57 for (d
= 0; d
< bytes
; d
+= 32) {
58 asm volatile("prefetchnta %0" : : "m" (dptr
[z0
][d
]));
59 asm volatile("vmovdqa %0,%%ymm2" : : "m" (dptr
[z0
][d
]));/* P[0] */
60 asm volatile("prefetchnta %0" : : "m" (dptr
[z0
-1][d
]));
61 asm volatile("vmovdqa %ymm2,%ymm4");/* Q[0] */
62 asm volatile("vmovdqa %0,%%ymm6" : : "m" (dptr
[z0
-1][d
]));
63 for (z
= z0
-2; z
>= 0; z
--) {
64 asm volatile("prefetchnta %0" : : "m" (dptr
[z
][d
]));
65 asm volatile("vpcmpgtb %ymm4,%ymm3,%ymm5");
66 asm volatile("vpaddb %ymm4,%ymm4,%ymm4");
67 asm volatile("vpand %ymm0,%ymm5,%ymm5");
68 asm volatile("vpxor %ymm5,%ymm4,%ymm4");
69 asm volatile("vpxor %ymm6,%ymm2,%ymm2");
70 asm volatile("vpxor %ymm6,%ymm4,%ymm4");
71 asm volatile("vmovdqa %0,%%ymm6" : : "m" (dptr
[z
][d
]));
73 asm volatile("vpcmpgtb %ymm4,%ymm3,%ymm5");
74 asm volatile("vpaddb %ymm4,%ymm4,%ymm4");
75 asm volatile("vpand %ymm0,%ymm5,%ymm5");
76 asm volatile("vpxor %ymm5,%ymm4,%ymm4");
77 asm volatile("vpxor %ymm6,%ymm2,%ymm2");
78 asm volatile("vpxor %ymm6,%ymm4,%ymm4");
80 asm volatile("vmovntdq %%ymm2,%0" : "=m" (p
[d
]));
81 asm volatile("vpxor %ymm2,%ymm2,%ymm2");
82 asm volatile("vmovntdq %%ymm4,%0" : "=m" (q
[d
]));
83 asm volatile("vpxor %ymm4,%ymm4,%ymm4");
86 asm volatile("sfence" : : : "memory");
90 const struct raid6_calls raid6_avx2x1
= {
91 raid6_avx21_gen_syndrome
,
92 NULL
, /* XOR not yet implemented */
95 1 /* Has cache hints */
99 * Unrolled-by-2 AVX2 implementation
101 static void raid6_avx22_gen_syndrome(int disks
, size_t bytes
, void **ptrs
)
103 u8
**dptr
= (u8
**)ptrs
;
107 z0
= disks
- 3; /* Highest data disk */
108 p
= dptr
[z0
+1]; /* XOR parity */
109 q
= dptr
[z0
+2]; /* RS syndrome */
113 asm volatile("vmovdqa %0,%%ymm0" : : "m" (raid6_avx2_constants
.x1d
[0]));
114 asm volatile("vpxor %ymm1,%ymm1,%ymm1"); /* Zero temp */
116 /* We uniformly assume a single prefetch covers at least 32 bytes */
117 for (d
= 0; d
< bytes
; d
+= 64) {
118 asm volatile("prefetchnta %0" : : "m" (dptr
[z0
][d
]));
119 asm volatile("prefetchnta %0" : : "m" (dptr
[z0
][d
+32]));
120 asm volatile("vmovdqa %0,%%ymm2" : : "m" (dptr
[z0
][d
]));/* P[0] */
121 asm volatile("vmovdqa %0,%%ymm3" : : "m" (dptr
[z0
][d
+32]));/* P[1] */
122 asm volatile("vmovdqa %ymm2,%ymm4"); /* Q[0] */
123 asm volatile("vmovdqa %ymm3,%ymm6"); /* Q[1] */
124 for (z
= z0
-1; z
>= 0; z
--) {
125 asm volatile("prefetchnta %0" : : "m" (dptr
[z
][d
]));
126 asm volatile("prefetchnta %0" : : "m" (dptr
[z
][d
+32]));
127 asm volatile("vpcmpgtb %ymm4,%ymm1,%ymm5");
128 asm volatile("vpcmpgtb %ymm6,%ymm1,%ymm7");
129 asm volatile("vpaddb %ymm4,%ymm4,%ymm4");
130 asm volatile("vpaddb %ymm6,%ymm6,%ymm6");
131 asm volatile("vpand %ymm0,%ymm5,%ymm5");
132 asm volatile("vpand %ymm0,%ymm7,%ymm7");
133 asm volatile("vpxor %ymm5,%ymm4,%ymm4");
134 asm volatile("vpxor %ymm7,%ymm6,%ymm6");
135 asm volatile("vmovdqa %0,%%ymm5" : : "m" (dptr
[z
][d
]));
136 asm volatile("vmovdqa %0,%%ymm7" : : "m" (dptr
[z
][d
+32]));
137 asm volatile("vpxor %ymm5,%ymm2,%ymm2");
138 asm volatile("vpxor %ymm7,%ymm3,%ymm3");
139 asm volatile("vpxor %ymm5,%ymm4,%ymm4");
140 asm volatile("vpxor %ymm7,%ymm6,%ymm6");
142 asm volatile("vmovntdq %%ymm2,%0" : "=m" (p
[d
]));
143 asm volatile("vmovntdq %%ymm3,%0" : "=m" (p
[d
+32]));
144 asm volatile("vmovntdq %%ymm4,%0" : "=m" (q
[d
]));
145 asm volatile("vmovntdq %%ymm6,%0" : "=m" (q
[d
+32]));
148 asm volatile("sfence" : : : "memory");
152 const struct raid6_calls raid6_avx2x2
= {
153 raid6_avx22_gen_syndrome
,
154 NULL
, /* XOR not yet implemented */
157 1 /* Has cache hints */
163 * Unrolled-by-4 AVX2 implementation
165 static void raid6_avx24_gen_syndrome(int disks
, size_t bytes
, void **ptrs
)
167 u8
**dptr
= (u8
**)ptrs
;
171 z0
= disks
- 3; /* Highest data disk */
172 p
= dptr
[z0
+1]; /* XOR parity */
173 q
= dptr
[z0
+2]; /* RS syndrome */
177 asm volatile("vmovdqa %0,%%ymm0" : : "m" (raid6_avx2_constants
.x1d
[0]));
178 asm volatile("vpxor %ymm1,%ymm1,%ymm1"); /* Zero temp */
179 asm volatile("vpxor %ymm2,%ymm2,%ymm2"); /* P[0] */
180 asm volatile("vpxor %ymm3,%ymm3,%ymm3"); /* P[1] */
181 asm volatile("vpxor %ymm4,%ymm4,%ymm4"); /* Q[0] */
182 asm volatile("vpxor %ymm6,%ymm6,%ymm6"); /* Q[1] */
183 asm volatile("vpxor %ymm10,%ymm10,%ymm10"); /* P[2] */
184 asm volatile("vpxor %ymm11,%ymm11,%ymm11"); /* P[3] */
185 asm volatile("vpxor %ymm12,%ymm12,%ymm12"); /* Q[2] */
186 asm volatile("vpxor %ymm14,%ymm14,%ymm14"); /* Q[3] */
188 for (d
= 0; d
< bytes
; d
+= 128) {
189 for (z
= z0
; z
>= 0; z
--) {
190 asm volatile("prefetchnta %0" : : "m" (dptr
[z
][d
]));
191 asm volatile("prefetchnta %0" : : "m" (dptr
[z
][d
+32]));
192 asm volatile("prefetchnta %0" : : "m" (dptr
[z
][d
+64]));
193 asm volatile("prefetchnta %0" : : "m" (dptr
[z
][d
+96]));
194 asm volatile("vpcmpgtb %ymm4,%ymm1,%ymm5");
195 asm volatile("vpcmpgtb %ymm6,%ymm1,%ymm7");
196 asm volatile("vpcmpgtb %ymm12,%ymm1,%ymm13");
197 asm volatile("vpcmpgtb %ymm14,%ymm1,%ymm15");
198 asm volatile("vpaddb %ymm4,%ymm4,%ymm4");
199 asm volatile("vpaddb %ymm6,%ymm6,%ymm6");
200 asm volatile("vpaddb %ymm12,%ymm12,%ymm12");
201 asm volatile("vpaddb %ymm14,%ymm14,%ymm14");
202 asm volatile("vpand %ymm0,%ymm5,%ymm5");
203 asm volatile("vpand %ymm0,%ymm7,%ymm7");
204 asm volatile("vpand %ymm0,%ymm13,%ymm13");
205 asm volatile("vpand %ymm0,%ymm15,%ymm15");
206 asm volatile("vpxor %ymm5,%ymm4,%ymm4");
207 asm volatile("vpxor %ymm7,%ymm6,%ymm6");
208 asm volatile("vpxor %ymm13,%ymm12,%ymm12");
209 asm volatile("vpxor %ymm15,%ymm14,%ymm14");
210 asm volatile("vmovdqa %0,%%ymm5" : : "m" (dptr
[z
][d
]));
211 asm volatile("vmovdqa %0,%%ymm7" : : "m" (dptr
[z
][d
+32]));
212 asm volatile("vmovdqa %0,%%ymm13" : : "m" (dptr
[z
][d
+64]));
213 asm volatile("vmovdqa %0,%%ymm15" : : "m" (dptr
[z
][d
+96]));
214 asm volatile("vpxor %ymm5,%ymm2,%ymm2");
215 asm volatile("vpxor %ymm7,%ymm3,%ymm3");
216 asm volatile("vpxor %ymm13,%ymm10,%ymm10");
217 asm volatile("vpxor %ymm15,%ymm11,%ymm11");
218 asm volatile("vpxor %ymm5,%ymm4,%ymm4");
219 asm volatile("vpxor %ymm7,%ymm6,%ymm6");
220 asm volatile("vpxor %ymm13,%ymm12,%ymm12");
221 asm volatile("vpxor %ymm15,%ymm14,%ymm14");
223 asm volatile("vmovntdq %%ymm2,%0" : "=m" (p
[d
]));
224 asm volatile("vpxor %ymm2,%ymm2,%ymm2");
225 asm volatile("vmovntdq %%ymm3,%0" : "=m" (p
[d
+32]));
226 asm volatile("vpxor %ymm3,%ymm3,%ymm3");
227 asm volatile("vmovntdq %%ymm10,%0" : "=m" (p
[d
+64]));
228 asm volatile("vpxor %ymm10,%ymm10,%ymm10");
229 asm volatile("vmovntdq %%ymm11,%0" : "=m" (p
[d
+96]));
230 asm volatile("vpxor %ymm11,%ymm11,%ymm11");
231 asm volatile("vmovntdq %%ymm4,%0" : "=m" (q
[d
]));
232 asm volatile("vpxor %ymm4,%ymm4,%ymm4");
233 asm volatile("vmovntdq %%ymm6,%0" : "=m" (q
[d
+32]));
234 asm volatile("vpxor %ymm6,%ymm6,%ymm6");
235 asm volatile("vmovntdq %%ymm12,%0" : "=m" (q
[d
+64]));
236 asm volatile("vpxor %ymm12,%ymm12,%ymm12");
237 asm volatile("vmovntdq %%ymm14,%0" : "=m" (q
[d
+96]));
238 asm volatile("vpxor %ymm14,%ymm14,%ymm14");
241 asm volatile("sfence" : : : "memory");
245 const struct raid6_calls raid6_avx2x4
= {
246 raid6_avx24_gen_syndrome
,
247 NULL
, /* XOR not yet implemented */
250 1 /* Has cache hints */
254 #endif /* CONFIG_AS_AVX2 */