Optimize saxpy for POWER10
authorRajalakshmi Srinivasaraghavan <rajis@linux.ibm.com>
Mon, 26 Oct 2020 18:24:59 +0000 (13:24 -0500)
committerRajalakshmi Srinivasaraghavan <rajis@linux.ibm.com>
Mon, 26 Oct 2020 18:24:59 +0000 (13:24 -0500)
This patch makes use of new POWER10 vector pair instructions for
loads and stores.

kernel/power/KERNEL.POWER10
kernel/power/saxpy_microk_power10.c [new file with mode: 0644]
kernel/power/saxpy_power10.c [new file with mode: 0644]

index 86df7e3..1e514fc 100644 (file)
@@ -141,7 +141,7 @@ DASUMKERNEL  = dasum.c
 CASUMKERNEL  = casum.c
 ZASUMKERNEL  = zasum.c
 #
-SAXPYKERNEL  = saxpy.c
+SAXPYKERNEL  = saxpy_power10.c
 DAXPYKERNEL  = daxpy_power10.c
 ifneq ($(GCCVERSIONGTEQ9),1)
 CAXPYKERNEL  = caxpy_power9.S
diff --git a/kernel/power/saxpy_microk_power10.c b/kernel/power/saxpy_microk_power10.c
new file mode 100644 (file)
index 0000000..6ede1dc
--- /dev/null
@@ -0,0 +1,181 @@
+/***************************************************************************
+Copyright (c) 2020, 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_8 1
+
+static void saxpy_kernel_64(long n, float *x, float *y, float alpha)
+{
+  __vector float t0 = {alpha, alpha,alpha, alpha};
+
+  __asm__
+    (
+
+       "dcbt           0, %2           \n\t"
+       "dcbt           0, %3           \n\t"
+
+       "lxvp           32, 0(%2)       \n\t"
+       "lxvp           34, 32(%2)      \n\t"
+       "lxvp           40, 64(%2)      \n\t"
+       "lxvp           42, 96(%2)      \n\t"
+       "lxvp           48, 128(%2)     \n\t"
+       "lxvp           50, 160(%2)     \n\t"
+       "lxvp           52, 192(%2)     \n\t"
+       "lxvp           54, 224(%2)     \n\t"
+
+       "lxvp           36, 0(%3)       \n\t"
+       "lxvp           38, 32(%3)      \n\t"
+       "lxvp           44, 64(%3)      \n\t"
+       "lxvp           46, 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"
+
+       "addi           %2, %2, 256     \n\t"
+
+       "addic.         %1, %1, -64     \n\t"
+       "ble            two%=           \n\t"
+
+       ".align 5                       \n"
+     "one%=:                           \n\t"
+
+       "xvmaddasp      36, 32, %x4     \n\t"
+       "xvmaddasp      37, 33, %x4     \n\t"
+
+       "lxvp           32, 0(%2)       \n\t"
+       "stxvp          36, 0(%3)       \n\t"
+
+       "xvmaddasp      38, 34, %x4     \n\t"
+       "xvmaddasp      39, 35, %x4     \n\t"
+
+       "lxvp           34, 32(%2)      \n\t"
+       "stxvp          38, 32(%3)      \n\t"
+
+       "lxvp           36, 256(%3)     \n\t"
+       "lxvp           38, 288(%3)     \n\t"
+
+       "xvmaddasp      44, 40, %x4     \n\t"
+       "xvmaddasp      45, 41, %x4     \n\t"
+
+       "lxvp           40, 64(%2)      \n\t"
+       "stxvp          44, 64(%3)      \n\t"
+
+       "xvmaddasp      46, 42, %x4     \n\t"
+       "xvmaddasp      47, 43, %x4     \n\t"
+
+       "lxvp           42, 96(%2)      \n\t"
+       "stxvp          46, 96(%3)      \n\t"
+
+       "lxvp           44, 320(%3)     \n\t"
+       "lxvp           46, 352(%3)     \n\t"
+
+       "xvmaddasp      56, 48, %x4     \n\t"
+       "xvmaddasp      57, 49, %x4     \n\t"
+
+       "lxvp           48, 128(%2)     \n\t"
+       "stxvp          56, 128(%3)     \n\t"
+
+       "xvmaddasp      58, 50, %x4     \n\t"
+       "xvmaddasp      59, 51, %x4     \n\t"
+
+       "lxvp           50, 160(%2)     \n\t"
+       "stxvp          58, 160(%3)     \n\t"
+
+       "lxvp           56, 384(%3)     \n\t"
+       "lxvp           58, 416(%3)     \n\t"
+
+       "xvmaddasp      60, 52, %x4     \n\t"
+       "xvmaddasp      61, 53, %x4     \n\t"
+
+       "lxvp           52, 192(%2)     \n\t"
+       "stxvp          60, 192(%3)     \n\t"
+
+       "xvmaddasp      62, 54, %x4     \n\t"
+       "xvmaddasp      63, 55, %x4     \n\t"
+
+       "lxvp           54, 224(%2)     \n\t"
+       "stxvp          62, 224(%3)     \n\t"
+
+       "lxvp           60, 448(%3)     \n\t"
+       "lxvp           62, 480(%3)     \n\t"
+
+       "addi           %2, %2, 256     \n\t"
+       "addi           %3, %3, 256     \n\t"
+
+       "addic.         %1, %1, -64     \n\t"
+       "bgt            one%=           \n"
+
+     "two%=:                           \n\t"
+
+       "xvmaddasp      36, 32, %x4     \n\t"
+       "xvmaddasp      37, 33, %x4     \n\t"
+       "xvmaddasp      38, 34, %x4     \n\t"
+       "xvmaddasp      39, 35, %x4     \n\t"
+
+       "xvmaddasp      44, 40, %x4     \n\t"
+       "xvmaddasp      45, 41, %x4     \n\t"
+       "xvmaddasp      46, 42, %x4     \n\t"
+       "xvmaddasp      47, 43, %x4     \n\t"
+
+       "xvmaddasp      56, 48, %x4     \n\t"
+       "xvmaddasp      57, 49, %x4     \n\t"
+       "xvmaddasp      58, 50, %x4     \n\t"
+       "xvmaddasp      59, 51, %x4     \n\t"
+
+       "xvmaddasp      60, 52, %x4     \n\t"
+       "xvmaddasp      61, 53, %x4     \n\t"
+       "xvmaddasp      62, 54, %x4     \n\t"
+       "xvmaddasp      63, 55, %x4     \n\t"
+       "stxvp          36, 0(%3)       \n\t"
+       "stxvp          38, 32(%3)      \n\t"
+       "stxvp          44, 64(%3)      \n\t"
+       "stxvp          46, 96(%3)      \n\t"
+       "stxvp          56, 128(%3)     \n\t"
+       "stxvp          58, 160(%3)     \n\t"
+       "stxvp          60, 192(%3)     \n\t"
+       "stxvp          62, 224(%3)     \n\t"
+
+     "#n=%1 x=%5=%2 y=%0=%3 t0=%x4\n"
+     :
+       "+m" (*y),
+       "+r" (n),       // 1
+       "+b" (x),       // 2
+       "+b" (y)                // 3
+     :
+       "wa" (t0),      // 4
+       "m" (*x)
+     :
+       "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"
+     );
+
+}
+
+
diff --git a/kernel/power/saxpy_power10.c b/kernel/power/saxpy_power10.c
new file mode 100644 (file)
index 0000000..8c7c223
--- /dev/null
@@ -0,0 +1,119 @@
+/***************************************************************************
+Copyright (c) 2020, 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(__VEC__) || defined(__ALTIVEC__)
+#include "saxpy_microk_power10.c"
+#endif
+
+#ifndef HAVE_KERNEL_8
+static void saxpy_kernel_64(BLASLONG n, FLOAT *x, FLOAT *y, FLOAT alpha)
+{
+       BLASLONG register i = 0;
+
+       while(i < n)
+        {
+              y[i]   += alpha * x[i];
+              y[i+1] += alpha * x[i+1];
+              y[i+2] += alpha * x[i+2];
+              y[i+3] += alpha * x[i+3];
+              y[i+4] += alpha * x[i+4];
+              y[i+5] += alpha * x[i+5];
+              y[i+6] += alpha * x[i+6];
+              y[i+7] += alpha * x[i+7];
+              i+=8 ;
+
+       }
+
+}
+#endif
+
+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;
+       BLASLONG ix=0,iy=0;
+
+       if ( n <= 0 )  return(0);
+
+       if ( (inc_x == 1) && (inc_y == 1) )
+       {
+
+               BLASLONG n1 = n & -64;
+
+               if ( n1 )
+                       saxpy_kernel_64(n1, x, y, da);
+
+               i = n1;
+               while(i < n)
+               {
+
+                       y[i] += da * x[i] ;
+                       i++ ;
+
+               }
+               return(0);
+
+
+       }
+
+       BLASLONG n1 = n & -4;
+
+       while(i < n1)
+       {
+
+               FLOAT m1      = da * x[ix] ;
+               FLOAT m2      = da * x[ix+inc_x] ;
+               FLOAT m3      = da * x[ix+2*inc_x] ;
+               FLOAT m4      = da * x[ix+3*inc_x] ;
+
+               y[iy]         += m1 ;
+               y[iy+inc_y]   += m2 ;
+               y[iy+2*inc_y] += m3 ;
+               y[iy+3*inc_y] += m4 ;
+
+               ix  += inc_x*4 ;
+               iy  += inc_y*4 ;
+               i+=4 ;
+
+       }
+
+       while(i < n)
+       {
+
+               y[iy] += da * x[ix] ;
+               ix  += inc_x ;
+               iy  += inc_y ;
+               i++ ;
+
+       }
+       return(0);
+
+}
+
+