Merge pull request #3852 from bartoldeman/sscal-opt
Add sscal.c + microkernels for Haswell, Zen, Skylake and newer.
This commit is contained in:
Martin Kroeker
GitHub
commit
4c12fc5f5d
|
|
@ -1,3 +1,4 @@
|
|||
SSCALKERNEL = sscal.c
|
||||
DSCALKERNEL = dscal.c
|
||||
CSCALKERNEL = cscal.c
|
||||
ZSCALKERNEL = zscal.c
|
||||
|
|
|
|||
|
|
@ -1,3 +1,4 @@
|
|||
SSCALKERNEL = sscal.c
|
||||
DSCALKERNEL = dscal.c
|
||||
CSCALKERNEL = cscal.c
|
||||
ZSCALKERNEL = zscal.c
|
||||
|
|
|
|||
|
|
@ -0,0 +1,196 @@
|
|||
/***************************************************************************
|
||||
Copyright (c) 2013 - 2022, The OpenBLAS Project
|
||||
All rights reserved.
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are
|
||||
met:
|
||||
1. Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
2. Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in
|
||||
the documentation and/or other materials provided with the
|
||||
distribution.
|
||||
3. Neither the name of the OpenBLAS project nor the names of
|
||||
its contributors may be used to endorse or promote products
|
||||
derived from this software without specific prior written permission.
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE OPENBLAS PROJECT OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
|
||||
USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*****************************************************************************/
|
||||
|
||||
#include "common.h"
|
||||
|
||||
#if defined(HASWELL) || defined(ZEN)
|
||||
#include "sscal_microk_haswell-2.c"
|
||||
#elif defined (SKYLAKEX) || defined (COOPERLAKE) || defined (SAPPHIRERAPIDS)
|
||||
#include "sscal_microk_skylakex-2.c"
|
||||
#endif
|
||||
|
||||
|
||||
#if !defined(HAVE_KERNEL_16)
|
||||
|
||||
static void sscal_kernel_16( BLASLONG n, FLOAT *da , FLOAT *x )
|
||||
{
|
||||
|
||||
BLASLONG i;
|
||||
FLOAT alpha = *da;
|
||||
|
||||
for( i=0; i<n; i+=8 )
|
||||
{
|
||||
x[0] *= alpha;
|
||||
x[1] *= alpha;
|
||||
x[2] *= alpha;
|
||||
x[3] *= alpha;
|
||||
x[4] *= alpha;
|
||||
x[5] *= alpha;
|
||||
x[6] *= alpha;
|
||||
x[7] *= alpha;
|
||||
x+=8;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
static void sscal_kernel_16_zero( BLASLONG n, FLOAT *alpha , FLOAT *x )
|
||||
{
|
||||
|
||||
BLASLONG i;
|
||||
for( i=0; i<n; i+=8 )
|
||||
{
|
||||
x[0] = 0.0;
|
||||
x[1] = 0.0;
|
||||
x[2] = 0.0;
|
||||
x[3] = 0.0;
|
||||
x[4] = 0.0;
|
||||
x[5] = 0.0;
|
||||
x[6] = 0.0;
|
||||
x[7] = 0.0;
|
||||
x+=8;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
static void sscal_kernel_inc_8(BLASLONG n, FLOAT *alpha, FLOAT *x, BLASLONG inc_x) __attribute__ ((noinline));
|
||||
|
||||
static void sscal_kernel_inc_8(BLASLONG n, FLOAT *alpha, FLOAT *x, BLASLONG inc_x)
|
||||
{
|
||||
|
||||
BLASLONG i;
|
||||
BLASLONG inc_x2 = 2 * inc_x;
|
||||
BLASLONG inc_x3 = inc_x2 + inc_x;
|
||||
FLOAT t0,t1,t2,t3;
|
||||
FLOAT da = alpha[0];
|
||||
|
||||
for ( i=0; i<n; i+=4 )
|
||||
{
|
||||
t0 = da * x[0];
|
||||
t1 = da * x[inc_x];
|
||||
t2 = da * x[inc_x2];
|
||||
t3 = da * x[inc_x3];
|
||||
|
||||
x[0] = t0;
|
||||
x[inc_x] = t1;
|
||||
x[inc_x2] = t2;
|
||||
x[inc_x3] = t3;
|
||||
|
||||
x+=4*inc_x;
|
||||
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT da, FLOAT *x, BLASLONG inc_x, FLOAT *y, BLASLONG inc_y, FLOAT *dummy, BLASLONG dummy2)
|
||||
{
|
||||
BLASLONG i=0,j=0;
|
||||
|
||||
if ( inc_x != 1 )
|
||||
{
|
||||
|
||||
if ( da == 0.0 )
|
||||
{
|
||||
|
||||
BLASLONG n1 = n & -2;
|
||||
|
||||
while(j < n1)
|
||||
{
|
||||
|
||||
x[i]=0.0;
|
||||
x[i+inc_x]=0.0;
|
||||
i += 2*inc_x ;
|
||||
j+=2;
|
||||
|
||||
}
|
||||
|
||||
while(j < n)
|
||||
{
|
||||
|
||||
x[i]=0.0;
|
||||
i += inc_x ;
|
||||
j++;
|
||||
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
|
||||
BLASLONG n1 = n & -8;
|
||||
if ( n1 > 0 )
|
||||
{
|
||||
sscal_kernel_inc_8(n1, &da, x, inc_x);
|
||||
i = n1 * inc_x;
|
||||
j = n1;
|
||||
}
|
||||
|
||||
while(j < n)
|
||||
{
|
||||
|
||||
x[i] *= da;
|
||||
i += inc_x ;
|
||||
j++;
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
return(0);
|
||||
}
|
||||
|
||||
BLASLONG n1 = n & -16;
|
||||
if ( n1 > 0 )
|
||||
{
|
||||
if ( da == 0.0 )
|
||||
sscal_kernel_16_zero(n1 , &da , x);
|
||||
else
|
||||
sscal_kernel_16(n1 , &da , x);
|
||||
}
|
||||
|
||||
if ( da == 0.0 )
|
||||
{
|
||||
for ( i=n1 ; i<n; i++ )
|
||||
{
|
||||
x[i] = 0.0;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
|
||||
for ( i=n1 ; i<n; i++ )
|
||||
{
|
||||
x[i] *= da;
|
||||
}
|
||||
}
|
||||
return(0);
|
||||
}
|
||||
|
||||
|
||||
|
|
@ -0,0 +1,180 @@
|
|||
/***************************************************************************
|
||||
Copyright (c) 2014-2022, The OpenBLAS Project
|
||||
All rights reserved.
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are
|
||||
met:
|
||||
1. Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
2. Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in
|
||||
the documentation and/or other materials provided with the
|
||||
distribution.
|
||||
3. Neither the name of the OpenBLAS project nor the names of
|
||||
its contributors may be used to endorse or promote products
|
||||
derived from this software without specific prior written permission.
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE OPENBLAS PROJECT OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
|
||||
USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*****************************************************************************/
|
||||
|
||||
#define HAVE_KERNEL_16 1
|
||||
|
||||
static void sscal_kernel_16( BLASLONG n, FLOAT *alpha, FLOAT *x) __attribute__ ((noinline));
|
||||
|
||||
static void sscal_kernel_16( BLASLONG n, FLOAT *alpha, FLOAT *x)
|
||||
{
|
||||
|
||||
|
||||
BLASLONG n1 = n >> 5 ;
|
||||
BLASLONG n2 = n & 16 ;
|
||||
|
||||
__asm__ __volatile__
|
||||
(
|
||||
"vbroadcastss (%2), %%ymm0 \n\t" // alpha
|
||||
|
||||
"addq $128, %1 \n\t"
|
||||
|
||||
"cmpq $0, %0 \n\t"
|
||||
"je 4f \n\t"
|
||||
|
||||
"vmulps -128(%1), %%ymm0, %%ymm4 \n\t"
|
||||
"vmulps -96(%1), %%ymm0, %%ymm5 \n\t"
|
||||
|
||||
"vmulps -64(%1), %%ymm0, %%ymm6 \n\t"
|
||||
"vmulps -32(%1), %%ymm0, %%ymm7 \n\t"
|
||||
|
||||
"subq $1 , %0 \n\t"
|
||||
"jz 2f \n\t"
|
||||
|
||||
".p2align 4 \n\t"
|
||||
"1: \n\t"
|
||||
// "prefetcht0 640(%1) \n\t"
|
||||
|
||||
"vmovups %%ymm4 ,-128(%1) \n\t"
|
||||
"vmovups %%ymm5 , -96(%1) \n\t"
|
||||
"vmulps 0(%1), %%ymm0, %%ymm4 \n\t"
|
||||
|
||||
// "prefetcht0 704(%1) \n\t"
|
||||
|
||||
"vmovups %%ymm6 , -64(%1) \n\t"
|
||||
"vmulps 32(%1), %%ymm0, %%ymm5 \n\t"
|
||||
"vmovups %%ymm7 , -32(%1) \n\t"
|
||||
|
||||
"vmulps 64(%1), %%ymm0, %%ymm6 \n\t"
|
||||
"vmulps 96(%1), %%ymm0, %%ymm7 \n\t"
|
||||
|
||||
|
||||
"addq $128, %1 \n\t"
|
||||
"subq $1 , %0 \n\t"
|
||||
"jnz 1b \n\t"
|
||||
|
||||
"2: \n\t"
|
||||
|
||||
"vmovups %%ymm4 ,-128(%1) \n\t"
|
||||
"vmovups %%ymm5 , -96(%1) \n\t"
|
||||
|
||||
"vmovups %%ymm6 , -64(%1) \n\t"
|
||||
"vmovups %%ymm7 , -32(%1) \n\t"
|
||||
|
||||
"addq $128, %1 \n\t"
|
||||
|
||||
"4: \n\t"
|
||||
|
||||
"cmpq $16 ,%3 \n\t"
|
||||
"jne 5f \n\t"
|
||||
|
||||
"vmulps -128(%1), %%ymm0, %%ymm4 \n\t"
|
||||
"vmulps -96(%1), %%ymm0, %%ymm5 \n\t"
|
||||
|
||||
"vmovups %%ymm4 ,-128(%1) \n\t"
|
||||
"vmovups %%ymm5 , -96(%1) \n\t"
|
||||
|
||||
"5: \n\t"
|
||||
|
||||
"vzeroupper \n\t"
|
||||
|
||||
:
|
||||
"+r" (n1), // 0
|
||||
"+r" (x) // 1
|
||||
:
|
||||
"r" (alpha), // 2
|
||||
"r" (n2) // 3
|
||||
: "cc",
|
||||
"%xmm0", "%xmm1", "%xmm2", "%xmm3",
|
||||
"%xmm4", "%xmm5", "%xmm6", "%xmm7",
|
||||
"%xmm8", "%xmm9", "%xmm10", "%xmm11",
|
||||
"%xmm12", "%xmm13", "%xmm14", "%xmm15",
|
||||
"memory"
|
||||
);
|
||||
|
||||
}
|
||||
|
||||
|
||||
static void sscal_kernel_16_zero( BLASLONG n, FLOAT *alpha, FLOAT *x) __attribute__ ((noinline));
|
||||
|
||||
static void sscal_kernel_16_zero( BLASLONG n, FLOAT *alpha, FLOAT *x)
|
||||
{
|
||||
|
||||
|
||||
BLASLONG n1 = n >> 5 ;
|
||||
BLASLONG n2 = n & 16 ;
|
||||
|
||||
__asm__ __volatile__
|
||||
(
|
||||
"vxorpd %%ymm0, %%ymm0 , %%ymm0 \n\t"
|
||||
|
||||
"addq $128, %1 \n\t"
|
||||
|
||||
"cmpq $0, %0 \n\t"
|
||||
"je 2f \n\t"
|
||||
|
||||
".p2align 4 \n\t"
|
||||
"1: \n\t"
|
||||
|
||||
"vmovups %%ymm0 ,-128(%1) \n\t"
|
||||
"vmovups %%ymm0 , -96(%1) \n\t"
|
||||
|
||||
"vmovups %%ymm0 , -64(%1) \n\t"
|
||||
"vmovups %%ymm0 , -32(%1) \n\t"
|
||||
|
||||
"addq $128, %1 \n\t"
|
||||
"subq $1 , %0 \n\t"
|
||||
"jnz 1b \n\t"
|
||||
|
||||
"2: \n\t"
|
||||
|
||||
"cmpq $16 ,%3 \n\t"
|
||||
"jne 4f \n\t"
|
||||
|
||||
"vmovups %%ymm0 ,-128(%1) \n\t"
|
||||
"vmovups %%ymm0 , -96(%1) \n\t"
|
||||
|
||||
"4: \n\t"
|
||||
|
||||
"vzeroupper \n\t"
|
||||
|
||||
:
|
||||
"+r" (n1), // 0
|
||||
"+r" (x) // 1
|
||||
:
|
||||
"r" (alpha), // 2
|
||||
"r" (n2) // 3
|
||||
: "cc",
|
||||
"%xmm0", "%xmm1", "%xmm2", "%xmm3",
|
||||
"%xmm4", "%xmm5", "%xmm6", "%xmm7",
|
||||
"%xmm8", "%xmm9", "%xmm10", "%xmm11",
|
||||
"%xmm12", "%xmm13", "%xmm14", "%xmm15",
|
||||
"memory"
|
||||
);
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
|
@ -0,0 +1,86 @@
|
|||
/***************************************************************************
|
||||
Copyright (c) 2014-2015, The OpenBLAS Project
|
||||
All rights reserved.
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are
|
||||
met:
|
||||
1. Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
2. Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in
|
||||
the documentation and/or other materials provided with the
|
||||
distribution.
|
||||
3. Neither the name of the OpenBLAS project nor the names of
|
||||
its contributors may be used to endorse or promote products
|
||||
derived from this software without specific prior written permission.
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE OPENBLAS PROJECT OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
|
||||
USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*****************************************************************************/
|
||||
|
||||
/* need a new enough GCC for avx512 support */
|
||||
#if (( defined(__GNUC__) && __GNUC__ > 6 && defined(__AVX2__)) || (defined(__clang__) && __clang_major__ >= 6))
|
||||
|
||||
#include <immintrin.h>
|
||||
|
||||
#define HAVE_KERNEL_16 1
|
||||
|
||||
static void sscal_kernel_16( BLASLONG n, FLOAT *alpha, FLOAT *x)
|
||||
{
|
||||
int i = 0;
|
||||
|
||||
#ifdef __AVX512CD__
|
||||
__m512 __alpha5 = _mm512_broadcastss_ps(_mm_load_ss(alpha));
|
||||
BLASLONG nn = n & -32;
|
||||
for (; i < nn; i += 32) {
|
||||
__m512 a = _mm512_loadu_ps(&x[i + 0]);
|
||||
__m512 b = _mm512_loadu_ps(&x[i + 16]);
|
||||
a *= __alpha5;
|
||||
b *= __alpha5;
|
||||
_mm512_storeu_ps(&x[i + 0], a);
|
||||
_mm512_storeu_ps(&x[i + 16], b);
|
||||
}
|
||||
for (; i < n; i += 16) {
|
||||
_mm512_storeu_ps(&x[i + 0], __alpha5 * _mm512_loadu_ps(&x[i + 0]));
|
||||
}
|
||||
#else
|
||||
__m256 __alpha = _mm256_broadcastss_ps(_mm_load_ss(alpha));
|
||||
for (; i < n; i += 16) {
|
||||
_mm256_storeu_ps(&x[i + 0], __alpha * _mm256_loadu_ps(&x[i + 0]));
|
||||
_mm256_storeu_ps(&x[i + 8], __alpha * _mm256_loadu_ps(&x[i + 8]));
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
static void sscal_kernel_16_zero( BLASLONG n, FLOAT *alpha, FLOAT *x)
|
||||
{
|
||||
int i = 0;
|
||||
|
||||
/* question to self: Why is this not just memset() */
|
||||
|
||||
#ifdef __AVX512CD__
|
||||
__m512 zero = _mm512_setzero_ps();
|
||||
for (; i < n; i += 16) {
|
||||
_mm512_storeu_ps(&x[i], zero);
|
||||
}
|
||||
#else
|
||||
__m256 zero = _mm256_setzero_ps();
|
||||
for (; i < n; i += 16) {
|
||||
_mm256_storeu_ps(&x[i + 0], zero);
|
||||
_mm256_storeu_ps(&x[i + 8], zero);
|
||||
}
|
||||
#endif
|
||||
|
||||
}
|
||||
|
||||
#else
|
||||
#include "sscal_microk_haswell-2.c"
|
||||
#endif
|
||||
Loading…
Reference in New Issue