Merge pull request #3055 from RajalakshmiSR/swapp10
Optimize swap function for POWER10
This commit is contained in:
Martin Kroeker
GitHub
commit
5d6209e1f9
|
|
@ -36,9 +36,11 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
#include "common.h"
|
#include "common.h"
|
||||||
|
|
||||||
|
|
||||||
#if defined(POWER8) || defined(POWER9) || defined(POWER10)
|
|
||||||
#if defined(__VEC__) || defined(__ALTIVEC__)
|
#if defined(__VEC__) || defined(__ALTIVEC__)
|
||||||
|
#if defined(POWER8) || defined(POWER9)
|
||||||
#include "cswap_microk_power8.c"
|
#include "cswap_microk_power8.c"
|
||||||
|
#elif defined(POWER10)
|
||||||
|
#include "cswap_microk_power10.c"
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,127 @@
|
||||||
|
/***************************************************************************
|
||||||
|
Copyright (c) 2021, 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.
|
||||||
|
*****************************************************************************/
|
||||||
|
|
||||||
|
#if defined(DOUBLE)
|
||||||
|
#define HAVE_KERNEL_16 1
|
||||||
|
static void zswap_kernel_16 (long n, double *x, double *y)
|
||||||
|
#else
|
||||||
|
#define HAVE_KERNEL_32 1
|
||||||
|
static void cswap_kernel_32 (long n, float *x, float *y)
|
||||||
|
#endif
|
||||||
|
{
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
".align 5 \n"
|
||||||
|
"one%=: \n\t"
|
||||||
|
"lxvp 32, 0(%4) \n\t"
|
||||||
|
"lxvp 34, 32(%4) \n\t"
|
||||||
|
"lxvp 36, 64(%4) \n\t"
|
||||||
|
"lxvp 38, 96(%4) \n\t"
|
||||||
|
|
||||||
|
"lxvp 40, 128(%4) \n\t"
|
||||||
|
"lxvp 42, 160(%4) \n\t"
|
||||||
|
"lxvp 44, 192(%4) \n\t"
|
||||||
|
"lxvp 46, 224(%4) \n\t"
|
||||||
|
|
||||||
|
"lxvp 48, 0(%3) \n\t"
|
||||||
|
"lxvp 50, 32(%3) \n\t"
|
||||||
|
"lxvp 52, 64(%3) \n\t"
|
||||||
|
"lxvp 54, 96(%3) \n\t"
|
||||||
|
|
||||||
|
"lxvp 56, 128(%3) \n\t"
|
||||||
|
"lxvp 58, 160(%3) \n\t"
|
||||||
|
"lxvp 60, 192(%3) \n\t"
|
||||||
|
"lxvp 62, 224(%3) \n\t"
|
||||||
|
|
||||||
|
|
||||||
|
"stxv 33, 0(%3) \n\t"
|
||||||
|
"stxv 32, 16(%3) \n\t"
|
||||||
|
"stxv 35, 32(%3) \n\t"
|
||||||
|
"stxv 34, 48(%3) \n\t"
|
||||||
|
"stxv 37, 64(%3) \n\t"
|
||||||
|
"stxv 36, 80(%3) \n\t"
|
||||||
|
"stxv 39, 96(%3) \n\t"
|
||||||
|
"stxv 38, 112(%3) \n\t"
|
||||||
|
|
||||||
|
"addi %3, %3, 128 \n\t"
|
||||||
|
|
||||||
|
"stxv 41, 0(%3) \n\t"
|
||||||
|
"stxv 40, 16(%3) \n\t"
|
||||||
|
"stxv 43, 32(%3) \n\t"
|
||||||
|
"stxv 42, 48(%3) \n\t"
|
||||||
|
"stxv 45, 64(%3) \n\t"
|
||||||
|
"stxv 44, 80(%3) \n\t"
|
||||||
|
"stxv 47, 96(%3) \n\t"
|
||||||
|
"stxv 46, 112(%3) \n\t"
|
||||||
|
|
||||||
|
"addi %3, %3, 128 \n\t"
|
||||||
|
|
||||||
|
"stxv 49, 0(%4) \n\t"
|
||||||
|
"stxv 48, 16(%4) \n\t"
|
||||||
|
"stxv 51, 32(%4) \n\t"
|
||||||
|
"stxv 50, 48(%4) \n\t"
|
||||||
|
"stxv 53, 64(%4) \n\t"
|
||||||
|
"stxv 52, 80(%4) \n\t"
|
||||||
|
"stxv 55, 96(%4) \n\t"
|
||||||
|
"stxv 54, 112(%4) \n\t"
|
||||||
|
|
||||||
|
"addi %4, %4, 128 \n\t"
|
||||||
|
|
||||||
|
"stxv 57, 0(%4) \n\t"
|
||||||
|
"stxv 56, 16(%4) \n\t"
|
||||||
|
"stxv 59, 32(%4) \n\t"
|
||||||
|
"stxv 58, 48(%4) \n\t"
|
||||||
|
"stxv 61, 64(%4) \n\t"
|
||||||
|
"stxv 60, 80(%4) \n\t"
|
||||||
|
"stxv 63, 96(%4) \n\t"
|
||||||
|
"stxv 62, 112(%4) \n\t"
|
||||||
|
|
||||||
|
"addi %4, %4, 128 \n\t"
|
||||||
|
|
||||||
|
#if defined(DOUBLE)
|
||||||
|
"addic. %2, %2, -16 \n\t"
|
||||||
|
#else
|
||||||
|
"addic. %2, %2, -32 \n\t"
|
||||||
|
#endif
|
||||||
|
"bgt one%= \n"
|
||||||
|
|
||||||
|
"#n=%2 x=%0=%3 y=%1=%4"
|
||||||
|
:
|
||||||
|
"+m" (*x),
|
||||||
|
"+m" (*y),
|
||||||
|
"+r" (n), // 2
|
||||||
|
"+b" (x), // 3
|
||||||
|
"+b" (y) // 4
|
||||||
|
:
|
||||||
|
:
|
||||||
|
"cr0",
|
||||||
|
"vs32","vs33","vs34","vs35","vs36","vs37","vs38","vs39",
|
||||||
|
"vs40","vs41","vs42","vs43","vs44","vs45","vs46","vs47",
|
||||||
|
"vs48","vs49","vs50","vs51","vs52","vs53","vs54","vs55",
|
||||||
|
"vs56","vs57","vs58","vs59","vs60","vs61","vs62","vs63"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
@ -35,9 +35,11 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
|
||||||
#include "common.h"
|
#include "common.h"
|
||||||
|
|
||||||
#if defined(POWER8) || defined(POWER9) || defined(POWER10)
|
|
||||||
#if defined(__VEC__) || defined(__ALTIVEC__)
|
#if defined(__VEC__) || defined(__ALTIVEC__)
|
||||||
|
#if defined(POWER8) || defined(POWER9)
|
||||||
#include "dswap_microk_power8.c"
|
#include "dswap_microk_power8.c"
|
||||||
|
#elif defined(POWER10)
|
||||||
|
#include "swap_microk_power10.c"
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
@ -115,12 +117,30 @@ int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT dummy3, FLOAT *x,
|
||||||
if ( (inc_x == 1) && (inc_y == 1 ))
|
if ( (inc_x == 1) && (inc_y == 1 ))
|
||||||
{
|
{
|
||||||
|
|
||||||
|
#if defined(POWER10)
|
||||||
|
if ( n >= 32 )
|
||||||
|
{
|
||||||
|
BLASLONG align = ((32 - ((uintptr_t)y & (uintptr_t)0x1F)) >> 3) & 0x3;
|
||||||
|
for (i = 0; i < align; i++) {
|
||||||
|
temp = y[i];
|
||||||
|
y[i] = x[i];
|
||||||
|
x[i] = temp;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
BLASLONG n1 = (n-i) & -32;
|
||||||
|
if ( n1 > 0 )
|
||||||
|
{
|
||||||
|
dswap_kernel_32(n1,&x[i], &y[i]);
|
||||||
|
i+=n1;
|
||||||
|
}
|
||||||
|
#else
|
||||||
BLASLONG n1 = n & -32;
|
BLASLONG n1 = n & -32;
|
||||||
if ( n1 > 0 )
|
if ( n1 > 0 )
|
||||||
{
|
{
|
||||||
dswap_kernel_32(n1, x, y);
|
dswap_kernel_32(n1, x, y);
|
||||||
i=n1;
|
i=n1;
|
||||||
}
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
while(i < n)
|
while(i < n)
|
||||||
{
|
{
|
||||||
|
|
|
||||||
|
|
@ -35,9 +35,11 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
|
||||||
#include "common.h"
|
#include "common.h"
|
||||||
|
|
||||||
#if defined(POWER8) || defined(POWER9) || defined(POWER10)
|
|
||||||
#if defined(__VEC__) || defined(__ALTIVEC__)
|
#if defined(__VEC__) || defined(__ALTIVEC__)
|
||||||
|
#if defined(POWER8) || defined(POWER9)
|
||||||
#include "sswap_microk_power8.c"
|
#include "sswap_microk_power8.c"
|
||||||
|
#elif defined(POWER10)
|
||||||
|
#include "swap_microk_power10.c"
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
@ -115,12 +117,30 @@ int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT dummy3, FLOAT *x,
|
||||||
if ( (inc_x == 1) && (inc_y == 1 ))
|
if ( (inc_x == 1) && (inc_y == 1 ))
|
||||||
{
|
{
|
||||||
|
|
||||||
|
#if defined(POWER10)
|
||||||
|
if ( n >= 64 )
|
||||||
|
{
|
||||||
|
BLASLONG align = ((32 - ((uintptr_t)y & (uintptr_t)0x1F)) >> 2) & 0x7;
|
||||||
|
for (i = 0; i < align; i++) {
|
||||||
|
temp = y[i];
|
||||||
|
y[i] = x[i];
|
||||||
|
x[i] = temp;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
BLASLONG n1 = (n-i) & -64;
|
||||||
|
if ( n1 > 0 )
|
||||||
|
{
|
||||||
|
sswap_kernel_32(n1,&x[i], &y[i]);
|
||||||
|
i+=n1;
|
||||||
|
}
|
||||||
|
#else
|
||||||
BLASLONG n1 = n & -32;
|
BLASLONG n1 = n & -32;
|
||||||
if ( n1 > 0 )
|
if ( n1 > 0 )
|
||||||
{
|
{
|
||||||
sswap_kernel_32(n1, x, y);
|
sswap_kernel_32(n1, x, y);
|
||||||
i=n1;
|
i=n1;
|
||||||
}
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
while(i < n)
|
while(i < n)
|
||||||
{
|
{
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,105 @@
|
||||||
|
/***************************************************************************
|
||||||
|
Copyright (c) 2021, 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_32 1
|
||||||
|
|
||||||
|
#if defined(DOUBLE)
|
||||||
|
static void dswap_kernel_32 (long n, double *x, double *y)
|
||||||
|
#else
|
||||||
|
static void sswap_kernel_32 (long n, float *x, float *y)
|
||||||
|
#endif
|
||||||
|
{
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
".align 5 \n"
|
||||||
|
"one%=: \n\t"
|
||||||
|
|
||||||
|
"lxvp 32, 0(%4) \n\t"
|
||||||
|
"lxvp 34, 32(%4) \n\t"
|
||||||
|
"lxvp 36, 64(%4) \n\t"
|
||||||
|
"lxvp 38, 96(%4) \n\t"
|
||||||
|
|
||||||
|
"lxvp 40, 128(%4) \n\t"
|
||||||
|
"lxvp 42, 160(%4) \n\t"
|
||||||
|
"lxvp 44, 192(%4) \n\t"
|
||||||
|
"lxvp 46, 224(%4) \n\t"
|
||||||
|
|
||||||
|
"lxvp 48, 0(%3) \n\t"
|
||||||
|
"lxvp 50, 32(%3) \n\t"
|
||||||
|
"lxvp 52, 64(%3) \n\t"
|
||||||
|
"lxvp 54, 96(%3) \n\t"
|
||||||
|
|
||||||
|
"lxvp 56, 128(%3) \n\t"
|
||||||
|
"lxvp 58, 160(%3) \n\t"
|
||||||
|
"lxvp 60, 192(%3) \n\t"
|
||||||
|
"lxvp 62, 224(%3) \n\t"
|
||||||
|
|
||||||
|
"stxvp 32, 0(%3) \n\t"
|
||||||
|
"stxvp 34, 32(%3) \n\t"
|
||||||
|
"stxvp 36, 64(%3) \n\t"
|
||||||
|
"stxvp 38, 96(%3) \n\t"
|
||||||
|
|
||||||
|
"stxvp 40, 128(%3) \n\t"
|
||||||
|
"stxvp 42, 160(%3) \n\t"
|
||||||
|
"stxvp 44, 192(%3) \n\t"
|
||||||
|
"stxvp 46, 224(%3) \n\t"
|
||||||
|
|
||||||
|
"stxvp 48, 0(%4) \n\t"
|
||||||
|
"stxvp 50, 32(%4) \n\t"
|
||||||
|
"stxvp 52, 64(%4) \n\t"
|
||||||
|
"stxvp 54, 96(%4) \n\t"
|
||||||
|
|
||||||
|
"stxvp 56, 128(%4) \n\t"
|
||||||
|
"stxvp 58, 160(%4) \n\t"
|
||||||
|
"stxvp 60, 192(%4) \n\t"
|
||||||
|
"stxvp 62, 224(%4) \n\t"
|
||||||
|
|
||||||
|
"addi %4, %4, 256 \n\t"
|
||||||
|
"addi %3, %3, 256 \n\t"
|
||||||
|
|
||||||
|
#if defined(DOUBLE)
|
||||||
|
"addic. %2, %2, -32 \n\t"
|
||||||
|
#else
|
||||||
|
"addic. %2, %2, -64 \n\t"
|
||||||
|
#endif
|
||||||
|
"bgt one%= \n"
|
||||||
|
|
||||||
|
"#n=%2 x=%0=%3 y=%1=%4"
|
||||||
|
:
|
||||||
|
"+m" (*x),
|
||||||
|
"+m" (*y),
|
||||||
|
"+r" (n), // 2
|
||||||
|
"+b" (x), // 3
|
||||||
|
"+b" (y) // 4
|
||||||
|
:
|
||||||
|
:
|
||||||
|
"cr0",
|
||||||
|
"vs32","vs33","vs34","vs35","vs36","vs37","vs38","vs39",
|
||||||
|
"vs40","vs41","vs42","vs43","vs44","vs45","vs46","vs47",
|
||||||
|
"vs48","vs49","vs50","vs51","vs52","vs53","vs54","vs55",
|
||||||
|
"vs56","vs57","vs58","vs59","vs60","vs61","vs62","vs63"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
@ -36,9 +36,11 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
#include "common.h"
|
#include "common.h"
|
||||||
|
|
||||||
|
|
||||||
#if defined(POWER8) || defined(POWER9) || defined(POWER10)
|
|
||||||
#if defined(__VEC__) || defined(__ALTIVEC__)
|
#if defined(__VEC__) || defined(__ALTIVEC__)
|
||||||
|
#if defined(POWER8) || defined(POWER9)
|
||||||
#include "zswap_microk_power8.c"
|
#include "zswap_microk_power8.c"
|
||||||
|
#elif defined(POWER10)
|
||||||
|
#include "cswap_microk_power10.c"
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue