Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 1 | /* -*- linux-c -*- ------------------------------------------------------- * |
| 2 | * |
| 3 | * Copyright 2002 H. Peter Anvin - All Rights Reserved |
| 4 | * |
| 5 | * This program is free software; you can redistribute it and/or modify |
| 6 | * it under the terms of the GNU General Public License as published by |
| 7 | * the Free Software Foundation, Inc., 53 Temple Place Ste 330, |
| 8 | * Bostom MA 02111-1307, USA; either version 2 of the License, or |
| 9 | * (at your option) any later version; incorporated herein by reference. |
| 10 | * |
| 11 | * ----------------------------------------------------------------------- */ |
| 12 | |
| 13 | /* |
| 14 | * raid6sse2.c |
| 15 | * |
| 16 | * SSE-2 implementation of RAID-6 syndrome functions |
| 17 | * |
| 18 | */ |
| 19 | |
| 20 | #if defined(__i386__) || defined(__x86_64__) |
| 21 | |
| 22 | #include "raid6.h" |
| 23 | #include "raid6x86.h" |
| 24 | |
| 25 | static const struct raid6_sse_constants { |
| 26 | u64 x1d[2]; |
| 27 | } raid6_sse_constants __attribute__((aligned(16))) = { |
| 28 | { 0x1d1d1d1d1d1d1d1dULL, 0x1d1d1d1d1d1d1d1dULL }, |
| 29 | }; |
| 30 | |
| 31 | static int raid6_have_sse2(void) |
| 32 | { |
| 33 | #ifdef __KERNEL__ |
| 34 | /* Not really boot_cpu but "all_cpus" */ |
| 35 | return boot_cpu_has(X86_FEATURE_MMX) && |
| 36 | boot_cpu_has(X86_FEATURE_FXSR) && |
| 37 | boot_cpu_has(X86_FEATURE_XMM) && |
| 38 | boot_cpu_has(X86_FEATURE_XMM2); |
| 39 | #else |
| 40 | /* User space test code */ |
| 41 | u32 features = cpuid_features(); |
| 42 | return ( (features & (15<<23)) == (15<<23) ); |
| 43 | #endif |
| 44 | } |
| 45 | |
| 46 | /* |
| 47 | * Plain SSE2 implementation |
| 48 | */ |
| 49 | static void raid6_sse21_gen_syndrome(int disks, size_t bytes, void **ptrs) |
| 50 | { |
| 51 | u8 **dptr = (u8 **)ptrs; |
| 52 | u8 *p, *q; |
| 53 | int d, z, z0; |
| 54 | raid6_sse_save_t sa; |
| 55 | |
| 56 | z0 = disks - 3; /* Highest data disk */ |
| 57 | p = dptr[z0+1]; /* XOR parity */ |
| 58 | q = dptr[z0+2]; /* RS syndrome */ |
| 59 | |
| 60 | raid6_before_sse2(&sa); |
| 61 | |
| 62 | asm volatile("movdqa %0,%%xmm0" : : "m" (raid6_sse_constants.x1d[0])); |
| 63 | asm volatile("pxor %xmm5,%xmm5"); /* Zero temp */ |
| 64 | |
| 65 | for ( d = 0 ; d < bytes ; d += 16 ) { |
| 66 | asm volatile("prefetchnta %0" : : "m" (dptr[z0][d])); |
| 67 | asm volatile("movdqa %0,%%xmm2" : : "m" (dptr[z0][d])); /* P[0] */ |
| 68 | asm volatile("prefetchnta %0" : : "m" (dptr[z0-1][d])); |
| 69 | asm volatile("movdqa %xmm2,%xmm4"); /* Q[0] */ |
| 70 | asm volatile("movdqa %0,%%xmm6" : : "m" (dptr[z0-1][d])); |
| 71 | for ( z = z0-2 ; z >= 0 ; z-- ) { |
| 72 | asm volatile("prefetchnta %0" : : "m" (dptr[z][d])); |
| 73 | asm volatile("pcmpgtb %xmm4,%xmm5"); |
| 74 | asm volatile("paddb %xmm4,%xmm4"); |
| 75 | asm volatile("pand %xmm0,%xmm5"); |
| 76 | asm volatile("pxor %xmm5,%xmm4"); |
| 77 | asm volatile("pxor %xmm5,%xmm5"); |
| 78 | asm volatile("pxor %xmm6,%xmm2"); |
| 79 | asm volatile("pxor %xmm6,%xmm4"); |
| 80 | asm volatile("movdqa %0,%%xmm6" : : "m" (dptr[z][d])); |
| 81 | } |
| 82 | asm volatile("pcmpgtb %xmm4,%xmm5"); |
| 83 | asm volatile("paddb %xmm4,%xmm4"); |
| 84 | asm volatile("pand %xmm0,%xmm5"); |
| 85 | asm volatile("pxor %xmm5,%xmm4"); |
| 86 | asm volatile("pxor %xmm5,%xmm5"); |
| 87 | asm volatile("pxor %xmm6,%xmm2"); |
| 88 | asm volatile("pxor %xmm6,%xmm4"); |
| 89 | |
| 90 | asm volatile("movntdq %%xmm2,%0" : "=m" (p[d])); |
| 91 | asm volatile("pxor %xmm2,%xmm2"); |
| 92 | asm volatile("movntdq %%xmm4,%0" : "=m" (q[d])); |
| 93 | asm volatile("pxor %xmm4,%xmm4"); |
| 94 | } |
| 95 | |
| 96 | raid6_after_sse2(&sa); |
| 97 | asm volatile("sfence" : : : "memory"); |
| 98 | } |
| 99 | |
| 100 | const struct raid6_calls raid6_sse2x1 = { |
| 101 | raid6_sse21_gen_syndrome, |
| 102 | raid6_have_sse2, |
| 103 | "sse2x1", |
| 104 | 1 /* Has cache hints */ |
| 105 | }; |
| 106 | |
| 107 | /* |
| 108 | * Unrolled-by-2 SSE2 implementation |
| 109 | */ |
| 110 | static void raid6_sse22_gen_syndrome(int disks, size_t bytes, void **ptrs) |
| 111 | { |
| 112 | u8 **dptr = (u8 **)ptrs; |
| 113 | u8 *p, *q; |
| 114 | int d, z, z0; |
| 115 | raid6_sse_save_t sa; |
| 116 | |
| 117 | z0 = disks - 3; /* Highest data disk */ |
| 118 | p = dptr[z0+1]; /* XOR parity */ |
| 119 | q = dptr[z0+2]; /* RS syndrome */ |
| 120 | |
| 121 | raid6_before_sse2(&sa); |
| 122 | |
| 123 | asm volatile("movdqa %0,%%xmm0" : : "m" (raid6_sse_constants.x1d[0])); |
| 124 | asm volatile("pxor %xmm5,%xmm5"); /* Zero temp */ |
| 125 | asm volatile("pxor %xmm7,%xmm7"); /* Zero temp */ |
| 126 | |
| 127 | /* We uniformly assume a single prefetch covers at least 32 bytes */ |
| 128 | for ( d = 0 ; d < bytes ; d += 32 ) { |
| 129 | asm volatile("prefetchnta %0" : : "m" (dptr[z0][d])); |
| 130 | asm volatile("movdqa %0,%%xmm2" : : "m" (dptr[z0][d])); /* P[0] */ |
| 131 | asm volatile("movdqa %0,%%xmm3" : : "m" (dptr[z0][d+16])); /* P[1] */ |
| 132 | asm volatile("movdqa %xmm2,%xmm4"); /* Q[0] */ |
| 133 | asm volatile("movdqa %xmm3,%xmm6"); /* Q[1] */ |
| 134 | for ( z = z0-1 ; z >= 0 ; z-- ) { |
| 135 | asm volatile("prefetchnta %0" : : "m" (dptr[z][d])); |
| 136 | asm volatile("pcmpgtb %xmm4,%xmm5"); |
| 137 | asm volatile("pcmpgtb %xmm6,%xmm7"); |
| 138 | asm volatile("paddb %xmm4,%xmm4"); |
| 139 | asm volatile("paddb %xmm6,%xmm6"); |
| 140 | asm volatile("pand %xmm0,%xmm5"); |
| 141 | asm volatile("pand %xmm0,%xmm7"); |
| 142 | asm volatile("pxor %xmm5,%xmm4"); |
| 143 | asm volatile("pxor %xmm7,%xmm6"); |
| 144 | asm volatile("movdqa %0,%%xmm5" : : "m" (dptr[z][d])); |
| 145 | asm volatile("movdqa %0,%%xmm7" : : "m" (dptr[z][d+16])); |
| 146 | asm volatile("pxor %xmm5,%xmm2"); |
| 147 | asm volatile("pxor %xmm7,%xmm3"); |
| 148 | asm volatile("pxor %xmm5,%xmm4"); |
| 149 | asm volatile("pxor %xmm7,%xmm6"); |
| 150 | asm volatile("pxor %xmm5,%xmm5"); |
| 151 | asm volatile("pxor %xmm7,%xmm7"); |
| 152 | } |
| 153 | asm volatile("movntdq %%xmm2,%0" : "=m" (p[d])); |
| 154 | asm volatile("movntdq %%xmm3,%0" : "=m" (p[d+16])); |
| 155 | asm volatile("movntdq %%xmm4,%0" : "=m" (q[d])); |
| 156 | asm volatile("movntdq %%xmm6,%0" : "=m" (q[d+16])); |
| 157 | } |
| 158 | |
| 159 | raid6_after_sse2(&sa); |
| 160 | asm volatile("sfence" : : : "memory"); |
| 161 | } |
| 162 | |
| 163 | const struct raid6_calls raid6_sse2x2 = { |
| 164 | raid6_sse22_gen_syndrome, |
| 165 | raid6_have_sse2, |
| 166 | "sse2x2", |
| 167 | 1 /* Has cache hints */ |
| 168 | }; |
| 169 | |
| 170 | #endif |
| 171 | |
| 172 | #ifdef __x86_64__ |
| 173 | |
| 174 | /* |
| 175 | * Unrolled-by-4 SSE2 implementation |
| 176 | */ |
| 177 | static void raid6_sse24_gen_syndrome(int disks, size_t bytes, void **ptrs) |
| 178 | { |
| 179 | u8 **dptr = (u8 **)ptrs; |
| 180 | u8 *p, *q; |
| 181 | int d, z, z0; |
| 182 | raid6_sse16_save_t sa; |
| 183 | |
| 184 | z0 = disks - 3; /* Highest data disk */ |
| 185 | p = dptr[z0+1]; /* XOR parity */ |
| 186 | q = dptr[z0+2]; /* RS syndrome */ |
| 187 | |
| 188 | raid6_before_sse16(&sa); |
| 189 | |
| 190 | asm volatile("movdqa %0,%%xmm0" :: "m" (raid6_sse_constants.x1d[0])); |
| 191 | asm volatile("pxor %xmm2,%xmm2"); /* P[0] */ |
| 192 | asm volatile("pxor %xmm3,%xmm3"); /* P[1] */ |
| 193 | asm volatile("pxor %xmm4,%xmm4"); /* Q[0] */ |
| 194 | asm volatile("pxor %xmm5,%xmm5"); /* Zero temp */ |
| 195 | asm volatile("pxor %xmm6,%xmm6"); /* Q[1] */ |
| 196 | asm volatile("pxor %xmm7,%xmm7"); /* Zero temp */ |
| 197 | asm volatile("pxor %xmm10,%xmm10"); /* P[2] */ |
| 198 | asm volatile("pxor %xmm11,%xmm11"); /* P[3] */ |
| 199 | asm volatile("pxor %xmm12,%xmm12"); /* Q[2] */ |
| 200 | asm volatile("pxor %xmm13,%xmm13"); /* Zero temp */ |
| 201 | asm volatile("pxor %xmm14,%xmm14"); /* Q[3] */ |
| 202 | asm volatile("pxor %xmm15,%xmm15"); /* Zero temp */ |
| 203 | |
| 204 | for ( d = 0 ; d < bytes ; d += 64 ) { |
| 205 | for ( z = z0 ; z >= 0 ; z-- ) { |
| 206 | /* The second prefetch seems to improve performance... */ |
| 207 | asm volatile("prefetchnta %0" :: "m" (dptr[z][d])); |
| 208 | asm volatile("prefetchnta %0" :: "m" (dptr[z][d+32])); |
| 209 | asm volatile("pcmpgtb %xmm4,%xmm5"); |
| 210 | asm volatile("pcmpgtb %xmm6,%xmm7"); |
| 211 | asm volatile("pcmpgtb %xmm12,%xmm13"); |
| 212 | asm volatile("pcmpgtb %xmm14,%xmm15"); |
| 213 | asm volatile("paddb %xmm4,%xmm4"); |
| 214 | asm volatile("paddb %xmm6,%xmm6"); |
| 215 | asm volatile("paddb %xmm12,%xmm12"); |
| 216 | asm volatile("paddb %xmm14,%xmm14"); |
| 217 | asm volatile("pand %xmm0,%xmm5"); |
| 218 | asm volatile("pand %xmm0,%xmm7"); |
| 219 | asm volatile("pand %xmm0,%xmm13"); |
| 220 | asm volatile("pand %xmm0,%xmm15"); |
| 221 | asm volatile("pxor %xmm5,%xmm4"); |
| 222 | asm volatile("pxor %xmm7,%xmm6"); |
| 223 | asm volatile("pxor %xmm13,%xmm12"); |
| 224 | asm volatile("pxor %xmm15,%xmm14"); |
| 225 | asm volatile("movdqa %0,%%xmm5" :: "m" (dptr[z][d])); |
| 226 | asm volatile("movdqa %0,%%xmm7" :: "m" (dptr[z][d+16])); |
| 227 | asm volatile("movdqa %0,%%xmm13" :: "m" (dptr[z][d+32])); |
| 228 | asm volatile("movdqa %0,%%xmm15" :: "m" (dptr[z][d+48])); |
| 229 | asm volatile("pxor %xmm5,%xmm2"); |
| 230 | asm volatile("pxor %xmm7,%xmm3"); |
| 231 | asm volatile("pxor %xmm13,%xmm10"); |
| 232 | asm volatile("pxor %xmm15,%xmm11"); |
| 233 | asm volatile("pxor %xmm5,%xmm4"); |
| 234 | asm volatile("pxor %xmm7,%xmm6"); |
| 235 | asm volatile("pxor %xmm13,%xmm12"); |
| 236 | asm volatile("pxor %xmm15,%xmm14"); |
| 237 | asm volatile("pxor %xmm5,%xmm5"); |
| 238 | asm volatile("pxor %xmm7,%xmm7"); |
| 239 | asm volatile("pxor %xmm13,%xmm13"); |
| 240 | asm volatile("pxor %xmm15,%xmm15"); |
| 241 | } |
| 242 | asm volatile("movntdq %%xmm2,%0" : "=m" (p[d])); |
| 243 | asm volatile("pxor %xmm2,%xmm2"); |
| 244 | asm volatile("movntdq %%xmm3,%0" : "=m" (p[d+16])); |
| 245 | asm volatile("pxor %xmm3,%xmm3"); |
| 246 | asm volatile("movntdq %%xmm10,%0" : "=m" (p[d+32])); |
| 247 | asm volatile("pxor %xmm10,%xmm10"); |
| 248 | asm volatile("movntdq %%xmm11,%0" : "=m" (p[d+48])); |
| 249 | asm volatile("pxor %xmm11,%xmm11"); |
| 250 | asm volatile("movntdq %%xmm4,%0" : "=m" (q[d])); |
| 251 | asm volatile("pxor %xmm4,%xmm4"); |
| 252 | asm volatile("movntdq %%xmm6,%0" : "=m" (q[d+16])); |
| 253 | asm volatile("pxor %xmm6,%xmm6"); |
| 254 | asm volatile("movntdq %%xmm12,%0" : "=m" (q[d+32])); |
| 255 | asm volatile("pxor %xmm12,%xmm12"); |
| 256 | asm volatile("movntdq %%xmm14,%0" : "=m" (q[d+48])); |
| 257 | asm volatile("pxor %xmm14,%xmm14"); |
| 258 | } |
| 259 | asm volatile("sfence" : : : "memory"); |
| 260 | raid6_after_sse16(&sa); |
| 261 | } |
| 262 | |
| 263 | const struct raid6_calls raid6_sse2x4 = { |
| 264 | raid6_sse24_gen_syndrome, |
| 265 | raid6_have_sse2, |
| 266 | "sse2x4", |
| 267 | 1 /* Has cache hints */ |
| 268 | }; |
| 269 | |
| 270 | #endif |