|
- /***************************************************************************
- Copyright (c) 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(DOUBLE)
- #define VSETVL(n) vsetvl_e32m2(n)
- #define FLOAT_V_T vfloat32m2_t
- #define VLEV_FLOAT vle32_v_f32m2
- #define VSEV_FLOAT vse32_v_f32m2
- #define VFMVVF_FLOAT vfmv_v_f_f32m2
- #define VFMACCVF_FLOAT vfmacc_vf_f32m2
- #define VFMULVF_FLOAT vfmul_vf_f32m2
- #else
- #define VSETVL(n) vsetvl_e64m2(n)
- #define FLOAT_V_T vfloat64m2_t
- #define VLEV_FLOAT vle64_v_f64m2
- #define VSEV_FLOAT vse64_v_f64m2
- #define VFMVVF_FLOAT vfmv_v_f_f64m2
- #define VFMACCVF_FLOAT vfmacc_vf_f64m2
- #define VFMULVF_FLOAT vfmul_vf_f64m2
- #endif
-
-
- // Optimizes the implementation in ../generic/trmmkernel_8x8.c
-
-
- int CNAME(BLASLONG bm,BLASLONG bn,BLASLONG bk,FLOAT alpha,FLOAT* ba,FLOAT* bb,FLOAT* C,BLASLONG ldc ,BLASLONG offset)
- {
- //fprintf(stderr, "%s, %s, bm=%4ld bn=%4ld bk=%4ld alpha=%f ldc=%ld\n", __FILE__, __FUNCTION__, bm, bn, bk, alpha, ldc);
-
- BLASLONG i,j,k;
- FLOAT *C0,*C1,*C2,*C3,*C4,*C5,*C6,*C7,*ptrba,*ptrbb;
-
- FLOAT_V_T va0, va1, va2, va3, va4, va5, va6, va7;
- FLOAT_V_T vres0, vres1, vres2, vres3, vres4, vres5, vres6, vres7;
- size_t vl;
-
- BLASLONG off, temp;
-
- #if !defined(LEFT)
- off = -offset;
- #else
- off = 0;
- #endif
- for (j = bn/8; j > 0; j--)
- {
- C0 = C;
- C1 = C0+ldc;
- C2 = C1+ldc;
- C3 = C2+ldc;
- C4 = C3+ldc;
- C5 = C4+ldc;
- C6 = C5+ldc;
- C7 = C6+ldc;
-
- #if defined(TRMMKERNEL) && defined(LEFT)
- off = offset;
- #endif
-
- ptrba = ba;
-
- for (i = bm; i > 0; i -= vl)
- {
- vl = VSETVL(i);
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- ptrbb = bb;
- #else
- ptrba += off*vl;
- ptrbb = bb + off*8;
- #endif
-
- vres0 = VFMVVF_FLOAT(0.0, vl);
- vres1 = VFMVVF_FLOAT(0.0, vl);
- vres2 = VFMVVF_FLOAT(0.0, vl);
- vres3 = VFMVVF_FLOAT(0.0, vl);
- vres4 = VFMVVF_FLOAT(0.0, vl);
- vres5 = VFMVVF_FLOAT(0.0, vl);
- vres6 = VFMVVF_FLOAT(0.0, vl);
- vres7 = VFMVVF_FLOAT(0.0, vl);
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- temp = bk-off;
- #elif defined(LEFT)
- temp = off+vl; // number of values in A
- #else
- temp = off+8; // number of values in B
- #endif
-
- for (k = temp/8; k > 0; k--) {
- va0 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
- va1 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va0, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va0, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va0, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va0, vl);
- vres4 = VFMACCVF_FLOAT(vres4, *(ptrbb + 4), va0, vl);
- vres5 = VFMACCVF_FLOAT(vres5, *(ptrbb + 5), va0, vl);
- vres6 = VFMACCVF_FLOAT(vres6, *(ptrbb + 6), va0, vl);
- vres7 = VFMACCVF_FLOAT(vres7, *(ptrbb + 7), va0, vl);
- ptrbb += 8;
- va2 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va1, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va1, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va1, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va1, vl);
- vres4 = VFMACCVF_FLOAT(vres4, *(ptrbb + 4), va1, vl);
- vres5 = VFMACCVF_FLOAT(vres5, *(ptrbb + 5), va1, vl);
- vres6 = VFMACCVF_FLOAT(vres6, *(ptrbb + 6), va1, vl);
- vres7 = VFMACCVF_FLOAT(vres7, *(ptrbb + 7), va1, vl);
- ptrbb += 8;
- va3 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va2, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va2, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va2, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va2, vl);
- vres4 = VFMACCVF_FLOAT(vres4, *(ptrbb + 4), va2, vl);
- vres5 = VFMACCVF_FLOAT(vres5, *(ptrbb + 5), va2, vl);
- vres6 = VFMACCVF_FLOAT(vres6, *(ptrbb + 6), va2, vl);
- vres7 = VFMACCVF_FLOAT(vres7, *(ptrbb + 7), va2, vl);
- ptrbb += 8;
- va4 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va3, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va3, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va3, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va3, vl);
- vres4 = VFMACCVF_FLOAT(vres4, *(ptrbb + 4), va3, vl);
- vres5 = VFMACCVF_FLOAT(vres5, *(ptrbb + 5), va3, vl);
- vres6 = VFMACCVF_FLOAT(vres6, *(ptrbb + 6), va3, vl);
- vres7 = VFMACCVF_FLOAT(vres7, *(ptrbb + 7), va3, vl);
- ptrbb += 8;
- va5 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va4, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va4, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va4, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va4, vl);
- vres4 = VFMACCVF_FLOAT(vres4, *(ptrbb + 4), va4, vl);
- vres5 = VFMACCVF_FLOAT(vres5, *(ptrbb + 5), va4, vl);
- vres6 = VFMACCVF_FLOAT(vres6, *(ptrbb + 6), va4, vl);
- vres7 = VFMACCVF_FLOAT(vres7, *(ptrbb + 7), va4, vl);
- ptrbb += 8;
- va6 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va5, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va5, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va5, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va5, vl);
- vres4 = VFMACCVF_FLOAT(vres4, *(ptrbb + 4), va5, vl);
- vres5 = VFMACCVF_FLOAT(vres5, *(ptrbb + 5), va5, vl);
- vres6 = VFMACCVF_FLOAT(vres6, *(ptrbb + 6), va5, vl);
- vres7 = VFMACCVF_FLOAT(vres7, *(ptrbb + 7), va5, vl);
- ptrbb += 8;
- va7 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va6, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va6, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va6, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va6, vl);
- vres4 = VFMACCVF_FLOAT(vres4, *(ptrbb + 4), va6, vl);
- vres5 = VFMACCVF_FLOAT(vres5, *(ptrbb + 5), va6, vl);
- vres6 = VFMACCVF_FLOAT(vres6, *(ptrbb + 6), va6, vl);
- vres7 = VFMACCVF_FLOAT(vres7, *(ptrbb + 7), va6, vl);
- ptrbb += 8;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va7, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va7, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va7, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va7, vl);
- vres4 = VFMACCVF_FLOAT(vres4, *(ptrbb + 4), va7, vl);
- vres5 = VFMACCVF_FLOAT(vres5, *(ptrbb + 5), va7, vl);
- vres6 = VFMACCVF_FLOAT(vres6, *(ptrbb + 6), va7, vl);
- vres7 = VFMACCVF_FLOAT(vres7, *(ptrbb + 7), va7, vl);
- ptrbb += 8;
- }
-
- for (k = temp&7; k > 0; k--) {
- va0 = VLEV_FLOAT(ptrba, vl); // M:8 (should be vlen);
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va0, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va0, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va0, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va0, vl);
- vres4 = VFMACCVF_FLOAT(vres4, *(ptrbb + 4), va0, vl);
- vres5 = VFMACCVF_FLOAT(vres5, *(ptrbb + 5), va0, vl);
- vres6 = VFMACCVF_FLOAT(vres6, *(ptrbb + 6), va0, vl);
- vres7 = VFMACCVF_FLOAT(vres7, *(ptrbb + 7), va0, vl);
-
- ptrbb += 8;
- ptrba += vl;
- }
-
- va0 = VFMULVF_FLOAT(vres0, alpha, vl);
- VSEV_FLOAT(C0, va0, vl);
-
- va1 = VFMULVF_FLOAT(vres1, alpha, vl);
- VSEV_FLOAT(C1, va1, vl);
-
- va2 = VFMULVF_FLOAT(vres2, alpha, vl);
- VSEV_FLOAT(C2, va2, vl);
-
- va3 = VFMULVF_FLOAT(vres3, alpha, vl);
- VSEV_FLOAT(C3, va3, vl);
-
- va4 = VFMULVF_FLOAT(vres4, alpha, vl);
- VSEV_FLOAT(C4, va4, vl);
-
- va5 = VFMULVF_FLOAT(vres5, alpha, vl);
- VSEV_FLOAT(C5, va5, vl);
-
- va6 = VFMULVF_FLOAT(vres6, alpha, vl);
- VSEV_FLOAT(C6, va6, vl);
-
- va7 = VFMULVF_FLOAT(vres7, alpha, vl);
- VSEV_FLOAT(C7, va7, vl);
-
-
- #if ( defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- temp = bk - off;
- #ifdef LEFT
- temp -= vl; // number of values in A
- #else
- temp -= 8; // number of values in B
- #endif
- ptrba += temp*vl;
- ptrbb += temp*8;
- #endif
-
- #ifdef LEFT
- off += vl; // number of values in A
- #endif
-
- C0 += vl;
- C1 += vl;
- C2 += vl;
- C3 += vl;
- C4 += vl;
- C5 += vl;
- C6 += vl;
- C7 += vl;
- }
-
- #if defined(TRMMKERNEL) && !defined(LEFT)
- off += 8;
- #endif
-
- bb += (bk<<3);
- C += (ldc<<3);
- }
-
- if (bn & 4)
- {
- C0 = C;
- C1 = C0+ldc;
- C2 = C1+ldc;
- C3 = C2+ldc;
-
- #if defined(TRMMKERNEL) && defined(LEFT)
- off = offset;
- #endif
- ptrba = ba;
-
- for (i = bm; i > 0; i -= vl)
- {
- vl = VSETVL(i);
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- ptrbb = bb;
- #else
- ptrba += off*vl;
- ptrbb = bb + off*4;
- #endif
-
- vres0 = VFMVVF_FLOAT(0.0, vl);
- vres1 = VFMVVF_FLOAT(0.0, vl);
- vres2 = VFMVVF_FLOAT(0.0, vl);
- vres3 = VFMVVF_FLOAT(0.0, vl);
-
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- temp = bk-off;
- #elif defined(LEFT)
- temp = off+vl; // number of values in A
- #else
- temp = off+4; // number of values in B
- #endif
-
- for (k = temp/8; k > 0; k--) {
- va0 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
- va1 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va0, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va0, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va0, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va0, vl);
- ptrbb += 4;
- va2 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va1, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va1, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va1, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va1, vl);
- ptrbb += 4;
- va3 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va2, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va2, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va2, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va2, vl);
- ptrbb += 4;
- va4 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va3, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va3, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va3, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va3, vl);
- ptrbb += 4;
- va5 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va4, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va4, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va4, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va4, vl);
- ptrbb += 4;
- va6 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va5, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va5, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va5, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va5, vl);
- ptrbb += 4;
- va7 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va6, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va6, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va6, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va6, vl);
- ptrbb += 4;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va7, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va7, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va7, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va7, vl);
- ptrbb += 4;
- }
-
- // K remainder
- for (k = temp&7; k > 0; k--) {
- va0 = VLEV_FLOAT(ptrba, vl);
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va0, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va0, vl);
- vres2 = VFMACCVF_FLOAT(vres2, *(ptrbb + 2), va0, vl);
- vres3 = VFMACCVF_FLOAT(vres3, *(ptrbb + 3), va0, vl);
-
- ptrbb += 4;
- ptrba += vl;
- }
-
- va0 = VFMULVF_FLOAT(vres0, alpha, vl);
- VSEV_FLOAT(C0, va0, vl);
-
- va1 = VFMULVF_FLOAT(vres1, alpha, vl);
- VSEV_FLOAT(C1, va1, vl);
-
- va2 = VFMULVF_FLOAT(vres2, alpha, vl);
- VSEV_FLOAT(C2, va2, vl);
-
- va3 = VFMULVF_FLOAT(vres3, alpha, vl);
- VSEV_FLOAT(C3, va3, vl);
-
- #if ( defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- temp = bk - off;
- #ifdef LEFT
- temp -= vl; // number of values in A
- #else
- temp -= 4; // number of values in B
- #endif
- ptrba += temp*vl;
- ptrbb += temp*4;
- #endif
-
- #ifdef LEFT
- off += vl; // number of values in A
- #endif
-
- C0 += vl;
- C1 += vl;
- C2 += vl;
- C3 += vl;
- }
-
- #if defined(TRMMKERNEL) && !defined(LEFT)
- off += 4;
- #endif
-
- bb += (bk<<2);
- C += (ldc<<2);
- }
-
- if (bn & 2)
- {
- C0 = C;
- C1 = C0+ldc;
-
- #if defined(TRMMKERNEL) && defined(LEFT)
- off = offset;
- #endif
-
- ptrba = ba;
-
- for (i = bm; i > 0; i -= vl)
- {
- vl = VSETVL(i);
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- ptrbb = bb;
- #else
- ptrba += off*vl;
- ptrbb = bb + off*2;
- #endif
-
- vres0 = VFMVVF_FLOAT(0.0, vl);
- vres1 = VFMVVF_FLOAT(0.0, vl);
-
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- temp = bk-off;
- #elif defined(LEFT)
- temp = off+vl; // number of values in A
- #else
- temp = off+2; // number of values in B
- #endif
-
- for (k = temp/8; k > 0; k--) {
- va0 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
- va1 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va0, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va0, vl);
- ptrbb += 2;
- va2 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va1, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va1, vl);
- ptrbb += 2;
- va3 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va2, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va2, vl);
- ptrbb += 2;
- va4 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va3, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va3, vl);
- ptrbb += 2;
- va5 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va4, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va4, vl);
- ptrbb += 2;
- va6 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va5, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va5, vl);
- ptrbb += 2;
- va7 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va6, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va6, vl);
- ptrbb += 2;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va7, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va7, vl);
- ptrbb += 2;
- }
-
- // K remainder
- for (k = temp&7; k > 0; k--) {
- va0 = VLEV_FLOAT(ptrba, vl);
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va0, vl);
- vres1 = VFMACCVF_FLOAT(vres1, *(ptrbb + 1), va0, vl);
-
- ptrbb += 2;
- ptrba += vl;
- }
- va0 = VFMULVF_FLOAT(vres0, alpha, vl);
- VSEV_FLOAT(C0, va0, vl);
-
- va1 = VFMULVF_FLOAT(vres1, alpha, vl);
- VSEV_FLOAT(C1, va1, vl);
-
- #if ( defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- temp = bk - off;
- #ifdef LEFT
- temp -= vl; // number of values in A
- #else
- temp -= 2; // number of values in B
- #endif
- ptrba += temp*vl;
- ptrbb += temp*2;
- #endif
-
- #ifdef LEFT
- off += vl; // number of values in A
- #endif
-
- C0 += vl;
- C1 += vl;
- }
-
- #if defined(TRMMKERNEL) && !defined(LEFT)
- off += 2;
- #endif
-
- bb += (bk<<1);
- C += (ldc<<1);
- }
-
- if (bn & 1)
- {
- C0 = C;
-
- #if defined(TRMMKERNEL) && defined(LEFT)
- off = offset;
- #endif
-
- ptrba = ba;
-
- for (i = bm; i > 0; i -= vl)
- {
- vl = VSETVL(i);
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- ptrbb = bb;
- #else
- ptrba += off*vl;
- ptrbb = bb + off*1;
- #endif
-
- vres0 = VFMVVF_FLOAT(0.0, vl);
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- temp = bk-off;
- #elif defined(LEFT)
- temp = off+vl; // number of values in A
- #else
- temp = off+1; // number of values in B
- #endif
-
- for (k = temp/8; k > 0; k--) {
- va0 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
- va1 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va0, vl);
- ptrbb += 1;
- va2 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va1, vl);
- ptrbb += 1;
- va3 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va2, vl);
- ptrbb += 1;
- va4 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va3, vl);
- ptrbb += 1;
- va5 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va4, vl);
- ptrbb += 1;
- va6 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va5, vl);
- ptrbb += 1;
- va7 = VLEV_FLOAT(ptrba, vl);
- ptrba += vl;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va6, vl);
- ptrbb += 1;
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va7, vl);
- ptrbb += 1;
- }
-
- // K remainder
- for (k = temp&7; k > 0; k--) {
- va0 = VLEV_FLOAT(ptrba, vl);
-
- vres0 = VFMACCVF_FLOAT(vres0, *(ptrbb + 0), va0, vl);
-
- ptrbb += 1;
- ptrba += vl;
- }
- va0 = VFMULVF_FLOAT(vres0, alpha, vl);
- VSEV_FLOAT(C0, va0, vl);
-
- #if ( defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- temp = bk - off;
- #ifdef LEFT
- temp -= vl; // number of values in A
- #else
- temp -= 1; // number of values in B
- #endif
- ptrba += temp*vl;
- ptrbb += temp*1;
- #endif
-
- #ifdef LEFT
- off += vl; // number of values in A
- #endif
-
- C0 += vl;
- }
-
- #if defined(TRMMKERNEL) && !defined(LEFT)
- off += 1;
- #endif
-
- bb += (bk);
- C += (ldc);
- }
- return 0;
- }
|