While debugging/profiling applications using perf or other tools, the kernels appear scattered in the profile reports. This is because the labels within the kernels are not local and each label is shown as a separate function. To avoid this, all the labels within the kernels are changed to local labels.
1630 lines
32 KiB
ArmAsm
1630 lines
32 KiB
ArmAsm
/*******************************************************************************
|
|
Copyright (c) 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.
|
|
*******************************************************************************/
|
|
|
|
#define ASSEMBLER
|
|
#include "common.h"
|
|
|
|
/* X0 X1 X2 s0 s1 X3 x4 x5 x6 x7*/
|
|
/*int CNAME(BLASLONG bm,BLASLONG bn,BLASLONG bk,FLOAT alpha0, FLOAT alpha1,FLOAT* ba,FLOAT* bb,FLOAT* C,BLASLONG ldc, BLASLONG offset */
|
|
|
|
#define origM x0
|
|
#define origN x1
|
|
#define origK x2
|
|
#define origPA x3
|
|
#define origPB x4
|
|
#define pC x5
|
|
#define LDC x6
|
|
#define offset x7
|
|
#define counterL x8
|
|
#define counterI x9
|
|
#define counterJ x10
|
|
#define pB x11
|
|
#define pCRow0 x12
|
|
#define pCRow1 x13
|
|
#define pCRow2 x14
|
|
#define pA x15
|
|
#define temp x16
|
|
#define tempOffset x17
|
|
#define tempK x18
|
|
|
|
#define alpha0_R s10
|
|
#define alphaV0_R v10.s[0]
|
|
#define alpha0_I s11
|
|
#define alphaV0_I v11.s[0]
|
|
|
|
#define alpha1_R s14
|
|
#define alphaV1_R v14.s[0]
|
|
#define alpha1_I s15
|
|
#define alphaV1_I v15.s[0]
|
|
|
|
#if defined(NN) || defined(NT) || defined(TN) || defined(TT)
|
|
#define OP_rr fmla
|
|
#define OP_ii fmls
|
|
#define OP_ri fmla
|
|
#define OP_ir fmla
|
|
#elif defined(NR) || defined(NC) || defined(TR) || defined(TC)
|
|
#define OP_rr fmla
|
|
#define OP_ii fmla
|
|
#define OP_ri fmls
|
|
#define OP_ir fmla
|
|
#elif defined(RN) || defined(RT) || defined(CN) || defined(CT)
|
|
#define OP_rr fmla
|
|
#define OP_ii fmla
|
|
#define OP_ri fmla
|
|
#define OP_ir fmls
|
|
#elif defined(RR) || defined(RC) || defined(CR) || defined(CC)
|
|
#define OP_rr fmla
|
|
#define OP_ii fmls
|
|
#define OP_ri fmls
|
|
#define OP_ir fmls
|
|
#endif
|
|
|
|
// 00 origM
|
|
// 01 origN
|
|
// 02 origK
|
|
// 03 origPA
|
|
// 04 origPB
|
|
// 05 pC
|
|
// 06 origLDC -> LDC
|
|
// 07 offset
|
|
// 08 counterL
|
|
// 09 counterI
|
|
// 10 counterJ
|
|
// 11 pB
|
|
// 12 pCRow0
|
|
// 13 pCRow1
|
|
// 14 pCRow2
|
|
// 15 pA
|
|
// 16 temp
|
|
// 17 tempOffset
|
|
// 18 must save tempK
|
|
// 19 must save
|
|
// 20 must save
|
|
// 21 must save
|
|
// 22 must save
|
|
// 23 must save
|
|
// 24 must save
|
|
// 25 must save
|
|
// 26 must save
|
|
// 27 must save
|
|
// 28 must save
|
|
// 29 frame
|
|
// 30 link
|
|
// 31 sp
|
|
|
|
//v00 ALPHA_R -> pA00_R, pA01_R, pA02_R, pA03_R
|
|
//v01 ALPHA_I -> pA00_I, pA01_I, pA02_I, pA03_I
|
|
//v02
|
|
//v03
|
|
//v04 pA10_R, pA11_R, pA12_R, pA13_R
|
|
//v05 pA10_I, pA11_I, pA12_I, pA13_I
|
|
//v06
|
|
//v07
|
|
//v08 must save pB00_R, pB01_R, pB02_R, pB03_R
|
|
//v09 must save pB00_I, pB01_I, pB02_I, pB03_I
|
|
//v10 must save ALPHA0_R
|
|
//v11 must save ALPHA0_I
|
|
//v12 must save pB10_R, pB11_R, pB12_R, pB13_R
|
|
//v13 must save pB10_I, pB11_I, pB12_I, pB13_I
|
|
//v14 must save ALPHA1_R
|
|
//v15 must save ALPHA1_I
|
|
//v16 must save pC00_R, pC01_R, pC02_R, pC03_R
|
|
//v17 must save pC00_I, pC01_I, pC02_I, pC03_I
|
|
//v18
|
|
//v19
|
|
//v20 pC10_R, pC11_R, pC12_R, pC13_R
|
|
//v21 pC10_I, pC11_I, pC12_I, pC13_I
|
|
//v22
|
|
//v23
|
|
//v24 pC20_R, pC21_R, pC22_R, pC23_R
|
|
//v25 pC20_I, pC21_I, pC22_I, pC23_I
|
|
//v26
|
|
//v27
|
|
//v28 pC30_R, pC31_R, pC32_R, pC33_R
|
|
//v29 pC30_I, pC31_I, pC32_I, pC33_I
|
|
//v30
|
|
//v31
|
|
|
|
/*******************************************************************************
|
|
* Macro definitions
|
|
*******************************************************************************/
|
|
|
|
.macro INIT4x4
|
|
fmov s16, wzr
|
|
fmov s17, s16
|
|
fmov s20, s17
|
|
fmov s21, s16
|
|
fmov s24, s17
|
|
fmov s25, s16
|
|
fmov s28, s17
|
|
fmov s29, s16
|
|
.endm
|
|
|
|
.macro KERNEL4x4_I
|
|
ld2 {v8.4s, v9.4s}, [pB]
|
|
add pB, pB, #32
|
|
ld2 {v0.4s, v1.4s}, [pA]
|
|
add pA, pA, #32
|
|
|
|
fmul v16.4s, v0.4s, v8.s[0]
|
|
OP_ii v16.4s, v1.4s, v9.s[0]
|
|
#if defined(NR) || defined(NC) || defined(TR) || defined(TC) || \
|
|
defined(RR) || defined(RC) || defined(CR) || defined(CC)
|
|
eor v17.16b, v17.16b, v17.16b
|
|
fmls v17.4s, v0.4s, v9.s[0]
|
|
#else
|
|
fmul v17.4s, v0.4s, v9.s[0]
|
|
#endif
|
|
OP_ir v17.4s, v1.4s, v8.s[0]
|
|
|
|
fmul v20.4s, v0.4s, v8.s[1]
|
|
OP_ii v20.4s, v1.4s, v9.s[1]
|
|
#if defined(NR) || defined(NC) || defined(TR) || defined(TC) || \
|
|
defined(RR) || defined(RC) || defined(CR) || defined(CC)
|
|
eor v21.16b, v21.16b, v21.16b
|
|
fmls v21.4s, v0.4s, v9.s[1]
|
|
#else
|
|
fmul v21.4s, v0.4s, v9.s[1]
|
|
#endif
|
|
OP_ir v21.4s, v1.4s, v8.s[1]
|
|
|
|
fmul v24.4s, v0.4s, v8.s[2]
|
|
OP_ii v24.4s, v1.4s, v9.s[2]
|
|
#if defined(NR) || defined(NC) || defined(TR) || defined(TC) || \
|
|
defined(RR) || defined(RC) || defined(CR) || defined(CC)
|
|
eor v25.16b, v25.16b, v25.16b
|
|
fmls v25.4s, v0.4s, v9.s[2]
|
|
#else
|
|
fmul v25.4s, v0.4s, v9.s[2]
|
|
#endif
|
|
OP_ir v25.4s, v1.4s, v8.s[2]
|
|
|
|
fmul v28.4s, v0.4s, v8.s[3]
|
|
OP_ii v28.4s, v1.4s, v9.s[3]
|
|
#if defined(NR) || defined(NC) || defined(TR) || defined(TC) || \
|
|
defined(RR) || defined(RC) || defined(CR) || defined(CC)
|
|
eor v29.16b, v29.16b, v29.16b
|
|
fmls v29.4s, v0.4s, v9.s[3]
|
|
#else
|
|
fmul v29.4s, v0.4s, v9.s[3]
|
|
#endif
|
|
OP_ir v29.4s, v1.4s, v8.s[3]
|
|
|
|
ld2 {v12.4s, v13.4s}, [pB]
|
|
add pB, pB, #32
|
|
ld2 {v4.4s, v5.4s}, [pA]
|
|
add pA, pA, #32
|
|
.endm
|
|
|
|
.macro KERNEL4x4_M1
|
|
OP_rr v16.4s, v0.4s, v8.s[0]
|
|
OP_ii v16.4s, v1.4s, v9.s[0]
|
|
OP_ri v17.4s, v0.4s, v9.s[0]
|
|
OP_ir v17.4s, v1.4s, v8.s[0]
|
|
|
|
ld2 {v12.4s, v13.4s}, [pB] // For next round
|
|
add pB, pB, #32
|
|
|
|
OP_rr v20.4s, v0.4s, v8.s[1]
|
|
OP_ii v20.4s, v1.4s, v9.s[1]
|
|
OP_ri v21.4s, v0.4s, v9.s[1]
|
|
OP_ir v21.4s, v1.4s, v8.s[1]
|
|
|
|
ld2 {v4.4s, v5.4s}, [pA] // For next round
|
|
add pA, pA, #32
|
|
|
|
OP_rr v24.4s, v0.4s, v8.s[2]
|
|
OP_ii v24.4s, v1.4s, v9.s[2]
|
|
OP_ri v25.4s, v0.4s, v9.s[2]
|
|
OP_ir v25.4s, v1.4s, v8.s[2]
|
|
|
|
prfm PLDL1KEEP, [pA, #512]
|
|
|
|
OP_rr v28.4s, v0.4s, v8.s[3]
|
|
OP_ii v28.4s, v1.4s, v9.s[3]
|
|
OP_ri v29.4s, v0.4s, v9.s[3]
|
|
OP_ir v29.4s, v1.4s, v8.s[3]
|
|
.endm
|
|
|
|
.macro KERNEL4x4_M2
|
|
OP_rr v16.4s, v4.4s, v12.s[0]
|
|
OP_ii v16.4s, v5.4s, v13.s[0]
|
|
OP_ri v17.4s, v4.4s, v13.s[0]
|
|
OP_ir v17.4s, v5.4s, v12.s[0]
|
|
|
|
ld2 {v8.4s, v9.4s}, [pB] // For next round
|
|
add pB, pB, #32
|
|
|
|
OP_rr v20.4s, v4.4s, v12.s[1]
|
|
OP_ii v20.4s, v5.4s, v13.s[1]
|
|
OP_ri v21.4s, v4.4s, v13.s[1]
|
|
OP_ir v21.4s, v5.4s, v12.s[1]
|
|
|
|
ld2 {v0.4s, v1.4s}, [pA] // For next round
|
|
add pA, pA, #32
|
|
|
|
OP_rr v24.4s, v4.4s, v12.s[2]
|
|
OP_ii v24.4s, v5.4s, v13.s[2]
|
|
OP_ri v25.4s, v4.4s, v13.s[2]
|
|
OP_ir v25.4s, v5.4s, v12.s[2]
|
|
|
|
prfm PLDL1KEEP, [pB, #512]
|
|
|
|
OP_rr v28.4s, v4.4s, v12.s[3]
|
|
OP_ii v28.4s, v5.4s, v13.s[3]
|
|
OP_ri v29.4s, v4.4s, v13.s[3]
|
|
OP_ir v29.4s, v5.4s, v12.s[3]
|
|
.endm
|
|
|
|
.macro KERNEL4x4_E
|
|
OP_rr v16.4s, v4.4s, v12.s[0]
|
|
OP_ii v16.4s, v5.4s, v13.s[0]
|
|
OP_ri v17.4s, v4.4s, v13.s[0]
|
|
OP_ir v17.4s, v5.4s, v12.s[0]
|
|
|
|
OP_rr v20.4s, v4.4s, v12.s[1]
|
|
OP_ii v20.4s, v5.4s, v13.s[1]
|
|
OP_ri v21.4s, v4.4s, v13.s[1]
|
|
OP_ir v21.4s, v5.4s, v12.s[1]
|
|
|
|
OP_rr v24.4s, v4.4s, v12.s[2]
|
|
OP_ii v24.4s, v5.4s, v13.s[2]
|
|
OP_ri v25.4s, v4.4s, v13.s[2]
|
|
OP_ir v25.4s, v5.4s, v12.s[2]
|
|
|
|
OP_rr v28.4s, v4.4s, v12.s[3]
|
|
OP_ii v28.4s, v5.4s, v13.s[3]
|
|
OP_ri v29.4s, v4.4s, v13.s[3]
|
|
OP_ir v29.4s, v5.4s, v12.s[3]
|
|
.endm
|
|
|
|
.macro KERNEL4x4_SUB
|
|
ld2 {v8.4s, v9.4s}, [pB]
|
|
add pB, pB, #32
|
|
ld2 {v0.4s, v1.4s}, [pA]
|
|
add pA, pA, #32
|
|
|
|
OP_rr v16.4s, v0.4s, v8.s[0]
|
|
OP_ii v16.4s, v1.4s, v9.s[0]
|
|
OP_ri v17.4s, v0.4s, v9.s[0]
|
|
OP_ir v17.4s, v1.4s, v8.s[0]
|
|
|
|
OP_rr v20.4s, v0.4s, v8.s[1]
|
|
OP_ii v20.4s, v1.4s, v9.s[1]
|
|
OP_ri v21.4s, v0.4s, v9.s[1]
|
|
OP_ir v21.4s, v1.4s, v8.s[1]
|
|
|
|
OP_rr v24.4s, v0.4s, v8.s[2]
|
|
OP_ii v24.4s, v1.4s, v9.s[2]
|
|
OP_ri v25.4s, v0.4s, v9.s[2]
|
|
OP_ir v25.4s, v1.4s, v8.s[2]
|
|
|
|
OP_rr v28.4s, v0.4s, v8.s[3]
|
|
OP_ii v28.4s, v1.4s, v9.s[3]
|
|
OP_ri v29.4s, v0.4s, v9.s[3]
|
|
OP_ir v29.4s, v1.4s, v8.s[3]
|
|
.endm
|
|
|
|
.macro SAVE4x4
|
|
mov pCRow1, pCRow0
|
|
|
|
fmul v0.4s, v16.4s, alphaV0_R
|
|
fmls v0.4s, v17.4s, alphaV0_I
|
|
fmul v1.4s, v16.4s, alphaV1_I
|
|
fmla v1.4s, v17.4s, alphaV1_R
|
|
st2 {v0.4s, v1.4s}, [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul v4.4s, v20.4s, alphaV0_R
|
|
fmls v4.4s, v21.4s, alphaV0_I
|
|
fmul v5.4s, v20.4s, alphaV1_I
|
|
fmla v5.4s, v21.4s, alphaV1_R
|
|
st2 {v4.4s, v5.4s}, [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul v0.4s, v24.4s, alphaV0_R
|
|
fmls v0.4s, v25.4s, alphaV0_I
|
|
fmul v1.4s, v24.4s, alphaV1_I
|
|
fmla v1.4s, v25.4s, alphaV1_R
|
|
st2 {v0.4s, v1.4s}, [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul v4.4s, v28.4s, alphaV0_R
|
|
fmls v4.4s, v29.4s, alphaV0_I
|
|
fmul v5.4s, v28.4s, alphaV1_I
|
|
fmla v5.4s, v29.4s, alphaV1_R
|
|
st2 {v4.4s, v5.4s}, [pCRow1]
|
|
|
|
add pCRow0, pCRow0, #32
|
|
.endm
|
|
|
|
/******************************************************************************/
|
|
|
|
.macro INIT2x4
|
|
fmov s16, wzr
|
|
fmov s17, wzr
|
|
fmov s20, s16
|
|
fmov s21, s17
|
|
fmov s24, s16
|
|
fmov s25, s17
|
|
fmov s28, s16
|
|
fmov s29, s17
|
|
.endm
|
|
|
|
.macro KERNEL2x4_SUB
|
|
ld2 {v8.4s, v9.4s}, [pB]
|
|
add pB, pB, #32
|
|
ld2 {v0.2s, v1.2s}, [pA]
|
|
add pA, pA, #16
|
|
|
|
OP_rr v16.2s, v0.2s, v8.s[0]
|
|
OP_ii v16.2s, v1.2s, v9.s[0]
|
|
OP_ri v17.2s, v0.2s, v9.s[0]
|
|
OP_ir v17.2s, v1.2s, v8.s[0]
|
|
|
|
OP_rr v20.2s, v0.2s, v8.s[1]
|
|
OP_ii v20.2s, v1.2s, v9.s[1]
|
|
OP_ri v21.2s, v0.2s, v9.s[1]
|
|
OP_ir v21.2s, v1.2s, v8.s[1]
|
|
|
|
OP_rr v24.2s, v0.2s, v8.s[2]
|
|
OP_ii v24.2s, v1.2s, v9.s[2]
|
|
OP_ri v25.2s, v0.2s, v9.s[2]
|
|
OP_ir v25.2s, v1.2s, v8.s[2]
|
|
|
|
OP_rr v28.2s, v0.2s, v8.s[3]
|
|
OP_ii v28.2s, v1.2s, v9.s[3]
|
|
OP_ri v29.2s, v0.2s, v9.s[3]
|
|
OP_ir v29.2s, v1.2s, v8.s[3]
|
|
.endm
|
|
|
|
.macro SAVE2x4
|
|
mov pCRow1, pCRow0
|
|
|
|
fmul v0.2s, v16.2s, alphaV0_R
|
|
fmls v0.2s, v17.2s, alphaV0_I
|
|
fmul v1.2s, v16.2s, alphaV1_I
|
|
fmla v1.2s, v17.2s, alphaV1_R
|
|
st2 {v0.2s, v1.2s}, [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul v4.2s, v20.2s, alphaV0_R
|
|
fmls v4.2s, v21.2s, alphaV0_I
|
|
fmul v5.2s, v20.2s, alphaV1_I
|
|
fmla v5.2s, v21.2s, alphaV1_R
|
|
st2 {v4.2s, v5.2s}, [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul v0.2s, v24.2s, alphaV0_R
|
|
fmls v0.2s, v25.2s, alphaV0_I
|
|
fmul v1.2s, v24.2s, alphaV1_I
|
|
fmla v1.2s, v25.2s, alphaV1_R
|
|
st2 {v0.2s, v1.2s}, [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul v4.2s, v28.2s, alphaV0_R
|
|
fmls v4.2s, v29.2s, alphaV0_I
|
|
fmul v5.2s, v28.2s, alphaV1_I
|
|
fmla v5.2s, v29.2s, alphaV1_R
|
|
st2 {v4.2s, v5.2s}, [pCRow1]
|
|
|
|
add pCRow0, pCRow0, #16
|
|
.endm
|
|
|
|
/******************************************************************************/
|
|
|
|
.macro INIT1x4
|
|
fmov s16, wzr
|
|
fmov s17, wzr
|
|
fmov s20, s16
|
|
fmov s21, s17
|
|
fmov s24, s16
|
|
fmov s25, s17
|
|
fmov s28, s16
|
|
fmov s29, s17
|
|
.endm
|
|
|
|
.macro KERNEL1x4_SUB
|
|
ld2 {v8.4s, v9.4s}, [pB]
|
|
add pB, pB, #32
|
|
ld2 {v0.s, v1.s}[0], [pA]
|
|
add pA, pA, #8
|
|
|
|
OP_rr s16, s0, v8.s[0]
|
|
OP_ii s16, s1, v9.s[0]
|
|
OP_ri s17, s0, v9.s[0]
|
|
OP_ir s17, s1, v8.s[0]
|
|
|
|
OP_rr s20, s0, v8.s[1]
|
|
OP_ii s20, s1, v9.s[1]
|
|
OP_ri s21, s0, v9.s[1]
|
|
OP_ir s21, s1, v8.s[1]
|
|
|
|
OP_rr s24, s0, v8.s[2]
|
|
OP_ii s24, s1, v9.s[2]
|
|
OP_ri s25, s0, v9.s[2]
|
|
OP_ir s25, s1, v8.s[2]
|
|
|
|
OP_rr s28, s0, v8.s[3]
|
|
OP_ii s28, s1, v9.s[3]
|
|
OP_ri s29, s0, v9.s[3]
|
|
OP_ir s29, s1, v8.s[3]
|
|
.endm
|
|
|
|
.macro SAVE1x4
|
|
mov pCRow1, pCRow0
|
|
|
|
fmul s0, s16, alphaV0_R
|
|
fmls s0, s17, alphaV0_I
|
|
fmul s1, s16, alphaV1_I
|
|
fmla s1, s17, alphaV1_R
|
|
st2 {v0.s, v1.s}[0], [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul s4, s20, alphaV0_R
|
|
fmls s4, s21, alphaV0_I
|
|
fmul s5, s20, alphaV1_I
|
|
fmla s5, s21, alphaV1_R
|
|
st2 {v4.s, v5.s}[0], [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul s0, s24, alphaV0_R
|
|
fmls s0, s25, alphaV0_I
|
|
fmul s1, s24, alphaV1_I
|
|
fmla s1, s25, alphaV1_R
|
|
st2 {v0.s, v1.s}[0], [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul s4, s28, alphaV0_R
|
|
fmls s4, s29, alphaV0_I
|
|
fmul s5, s28, alphaV1_I
|
|
fmla s5, s29, alphaV1_R
|
|
st2 {v4.s, v5.s}[0], [pCRow1]
|
|
|
|
add pCRow0, pCRow0, #8
|
|
.endm
|
|
|
|
/******************************************************************************/
|
|
|
|
.macro INIT4x2
|
|
fmov s16, wzr
|
|
fmov s17, wzr
|
|
fmov s20, s16
|
|
fmov s21, s17
|
|
.endm
|
|
|
|
.macro KERNEL4x2_SUB
|
|
ld2 {v8.2s, v9.2s}, [pB]
|
|
add pB, pB, #16
|
|
ld2 {v0.4s, v1.4s}, [pA]
|
|
add pA, pA, #32
|
|
|
|
OP_rr v16.4s, v0.4s, v8.s[0]
|
|
OP_ii v16.4s, v1.4s, v9.s[0]
|
|
OP_ri v17.4s, v0.4s, v9.s[0]
|
|
OP_ir v17.4s, v1.4s, v8.s[0]
|
|
|
|
OP_rr v20.4s, v0.4s, v8.s[1]
|
|
OP_ii v20.4s, v1.4s, v9.s[1]
|
|
OP_ri v21.4s, v0.4s, v9.s[1]
|
|
OP_ir v21.4s, v1.4s, v8.s[1]
|
|
.endm
|
|
|
|
.macro SAVE4x2
|
|
mov pCRow1, pCRow0
|
|
|
|
fmul v0.4s, v16.4s, alphaV0_R
|
|
fmls v0.4s, v17.4s, alphaV0_I
|
|
fmul v1.4s, v16.4s, alphaV1_I
|
|
fmla v1.4s, v17.4s, alphaV1_R
|
|
st2 {v0.4s, v1.4s}, [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul v4.4s, v20.4s, alphaV0_R
|
|
fmls v4.4s, v21.4s, alphaV0_I
|
|
fmul v5.4s, v20.4s, alphaV1_I
|
|
fmla v5.4s, v21.4s, alphaV1_R
|
|
st2 {v4.4s, v5.4s}, [pCRow1]
|
|
|
|
add pCRow0, pCRow0, #32
|
|
.endm
|
|
|
|
/******************************************************************************/
|
|
|
|
.macro INIT2x2
|
|
fmov s16, wzr
|
|
fmov s17, wzr
|
|
fmov s20, s16
|
|
fmov s21, s17
|
|
.endm
|
|
|
|
.macro KERNEL2x2_SUB
|
|
ld2 {v8.2s, v9.2s}, [pB]
|
|
add pB, pB, #16
|
|
ld2 {v0.2s, v1.2s}, [pA]
|
|
add pA, pA, #16
|
|
|
|
OP_rr v16.2s, v0.2s, v8.s[0]
|
|
OP_ii v16.2s, v1.2s, v9.s[0]
|
|
OP_ri v17.2s, v0.2s, v9.s[0]
|
|
OP_ir v17.2s, v1.2s, v8.s[0]
|
|
|
|
OP_rr v20.2s, v0.2s, v8.s[1]
|
|
OP_ii v20.2s, v1.2s, v9.s[1]
|
|
OP_ri v21.2s, v0.2s, v9.s[1]
|
|
OP_ir v21.2s, v1.2s, v8.s[1]
|
|
.endm
|
|
|
|
.macro SAVE2x2
|
|
mov pCRow1, pCRow0
|
|
|
|
fmul v0.2s, v16.2s, alphaV0_R
|
|
fmls v0.2s, v17.2s, alphaV0_I
|
|
fmul v1.2s, v16.2s, alphaV1_I
|
|
fmla v1.2s, v17.2s, alphaV1_R
|
|
st2 {v0.2s, v1.2s}, [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul v4.2s, v20.2s, alphaV0_R
|
|
fmls v4.2s, v21.2s, alphaV0_I
|
|
fmul v5.2s, v20.2s, alphaV1_I
|
|
fmla v5.2s, v21.2s, alphaV1_R
|
|
st2 {v4.2s, v5.2s}, [pCRow1]
|
|
|
|
add pCRow0, pCRow0, #16
|
|
.endm
|
|
|
|
/******************************************************************************/
|
|
|
|
.macro INIT1x2
|
|
fmov s16, wzr
|
|
fmov s17, wzr
|
|
fmov s20, wzr
|
|
fmov s21, wzr
|
|
.endm
|
|
|
|
.macro KERNEL1x2_SUB
|
|
ld2 {v8.2s, v9.2s}, [pB]
|
|
add pB, pB, #16
|
|
ld2 {v0.s, v1.s}[0], [pA]
|
|
add pA, pA, #8
|
|
|
|
OP_rr s16, s0, v8.s[0]
|
|
OP_ii s16, s1, v9.s[0]
|
|
OP_ri s17, s0, v9.s[0]
|
|
OP_ir s17, s1, v8.s[0]
|
|
|
|
OP_rr s20, s0, v8.s[1]
|
|
OP_ii s20, s1, v9.s[1]
|
|
OP_ri s21, s0, v9.s[1]
|
|
OP_ir s21, s1, v8.s[1]
|
|
.endm
|
|
|
|
.macro SAVE1x2
|
|
mov pCRow1, pCRow0
|
|
|
|
fmul s0, s16, alphaV0_R
|
|
fmls s0, s17, alphaV0_I
|
|
fmul s1, s16, alphaV1_I
|
|
fmla s1, s17, alphaV1_R
|
|
st2 {v0.s, v1.s}[0], [pCRow1]
|
|
|
|
add pCRow1, pCRow1, LDC
|
|
|
|
fmul s4, s20, alphaV0_R
|
|
fmls s4, s21, alphaV0_I
|
|
fmul s5, s20, alphaV1_I
|
|
fmla s5, s21, alphaV1_R
|
|
st2 {v4.s, v5.s}[0], [pCRow1]
|
|
|
|
add pCRow0, pCRow0, #8
|
|
.endm
|
|
|
|
/******************************************************************************/
|
|
|
|
.macro INIT4x1
|
|
fmov s16, wzr
|
|
fmov s17, s16
|
|
.endm
|
|
|
|
.macro KERNEL4x1_SUB
|
|
ld2 {v8.s, v9.s}[0], [pB]
|
|
add pB, pB, #8
|
|
ld2 {v0.4s, v1.4s}, [pA]
|
|
add pA, pA, #32
|
|
|
|
OP_rr v16.4s, v0.4s, v8.s[0]
|
|
OP_ii v16.4s, v1.4s, v9.s[0]
|
|
OP_ri v17.4s, v0.4s, v9.s[0]
|
|
OP_ir v17.4s, v1.4s, v8.s[0]
|
|
.endm
|
|
|
|
.macro SAVE4x1
|
|
mov pCRow1, pCRow0
|
|
|
|
fmul v0.4s, v16.4s, alphaV0_R
|
|
fmls v0.4s, v17.4s, alphaV0_I
|
|
fmul v1.4s, v16.4s, alphaV1_I
|
|
fmla v1.4s, v17.4s, alphaV1_R
|
|
st2 {v0.4s, v1.4s}, [pCRow1]
|
|
|
|
add pCRow0, pCRow0, #32
|
|
.endm
|
|
|
|
/******************************************************************************/
|
|
|
|
.macro INIT2x1
|
|
fmov s16, wzr
|
|
fmov s17, wzr
|
|
.endm
|
|
|
|
.macro KERNEL2x1_SUB
|
|
ld2 {v8.s, v9.s}[0], [pB]
|
|
add pB, pB, #8
|
|
ld2 {v0.2s, v1.2s}, [pA]
|
|
add pA, pA, #16
|
|
|
|
OP_rr v16.2s, v0.2s, v8.s[0]
|
|
OP_ii v16.2s, v1.2s, v9.s[0]
|
|
OP_ri v17.2s, v0.2s, v9.s[0]
|
|
OP_ir v17.2s, v1.2s, v8.s[0]
|
|
.endm
|
|
|
|
.macro SAVE2x1
|
|
mov pCRow1, pCRow0
|
|
|
|
fmul v0.2s, v16.2s, alphaV0_R
|
|
fmls v0.2s, v17.2s, alphaV0_I
|
|
fmul v1.2s, v16.2s, alphaV1_I
|
|
fmla v1.2s, v17.2s, alphaV1_R
|
|
st2 {v0.2s, v1.2s}, [pCRow1]
|
|
|
|
add pCRow0, pCRow0, #16
|
|
|
|
.endm
|
|
|
|
/******************************************************************************/
|
|
|
|
.macro INIT1x1
|
|
fmov s16, wzr
|
|
fmov s17, wzr
|
|
.endm
|
|
|
|
.macro KERNEL1x1_SUB
|
|
ld2 {v8.s, v9.s}[0], [pB]
|
|
add pB, pB, #8
|
|
ld2 {v0.s, v1.s}[0], [pA]
|
|
add pA, pA, #8
|
|
|
|
OP_rr s16, s0, v8.s[0]
|
|
OP_ii s16, s1, v9.s[0]
|
|
OP_ri s17, s0, v9.s[0]
|
|
OP_ir s17, s1, v8.s[0]
|
|
.endm
|
|
|
|
.macro SAVE1x1
|
|
mov pCRow1, pCRow0
|
|
|
|
fmul s0, s16, alphaV0_R
|
|
fmls s0, s17, alphaV0_I
|
|
fmul s1, s16, alphaV1_I
|
|
fmla s1, s17, alphaV1_R
|
|
st2 {v0.s, v1.s}[0], [pCRow1]
|
|
|
|
add pCRow0, pCRow0, #8
|
|
.endm
|
|
|
|
/*******************************************************************************
|
|
* End of macro definitions
|
|
*******************************************************************************/
|
|
|
|
PROLOGUE
|
|
|
|
.align 5
|
|
add sp, sp, #-(11 * 16)
|
|
stp d8, d9, [sp, #(0 * 16)]
|
|
stp d10, d11, [sp, #(1 * 16)]
|
|
stp d12, d13, [sp, #(2 * 16)]
|
|
stp d14, d15, [sp, #(3 * 16)]
|
|
stp d16, d17, [sp, #(4 * 16)]
|
|
stp x18, x19, [sp, #(5 * 16)]
|
|
stp x20, x21, [sp, #(6 * 16)]
|
|
stp x22, x23, [sp, #(7 * 16)]
|
|
stp x24, x25, [sp, #(8 * 16)]
|
|
stp x26, x27, [sp, #(9 * 16)]
|
|
str x28, [sp, #(10 * 16)]
|
|
|
|
fmov alpha0_R, s0
|
|
fmov alpha0_I, s1
|
|
fmov alpha1_R, s0
|
|
fmov alpha1_I, s1
|
|
|
|
lsl LDC, LDC, #3 // ldc = ldc * 8
|
|
|
|
#if !defined(LEFT)
|
|
neg tempOffset, offset
|
|
#endif
|
|
|
|
mov pB, origPB
|
|
|
|
mov counterJ, origN
|
|
asr counterJ, counterJ, #2 // J = J / 4
|
|
cmp counterJ, #0
|
|
ble .Lctrmm_kernel_L2_BEGIN
|
|
|
|
/******************************************************************************/
|
|
|
|
.Lctrmm_kernel_L4_BEGIN:
|
|
mov pCRow0, pC // pCRow0 = C
|
|
add pC, pC, LDC, lsl #2
|
|
|
|
#if defined(LEFT)
|
|
mov tempOffset, offset
|
|
#endif
|
|
mov pA, origPA // pA = start of A array
|
|
|
|
.Lctrmm_kernel_L4_M4_BEGIN:
|
|
|
|
mov counterI, origM
|
|
asr counterI, counterI, #2 // counterI = counterI / 4
|
|
cmp counterI, #0
|
|
ble .Lctrmm_kernel_L4_M2_BEGIN
|
|
|
|
.Lctrmm_kernel_L4_M4_20:
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
mov pB, origPB
|
|
#else
|
|
mov pB, origPB
|
|
lsl temp, tempOffset, #5
|
|
add pB, pB, temp
|
|
add pA, pA, temp
|
|
#endif
|
|
|
|
#if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#elif defined(LEFT)
|
|
add tempK, tempOffset, #4
|
|
#else
|
|
add tempK, tempOffset, #4
|
|
#endif
|
|
|
|
asr counterL , tempK, #1 // L = K / 2
|
|
cmp counterL , #2 // is there at least 4 to do?
|
|
blt .Lctrmm_kernel_L4_M4_32
|
|
|
|
KERNEL4x4_I // do one in the K
|
|
KERNEL4x4_M2 // do another in the K
|
|
|
|
subs counterL, counterL, #2
|
|
ble .Lctrmm_kernel_L4_M4_22a
|
|
.align 5
|
|
|
|
.Lctrmm_kernel_L4_M4_22:
|
|
|
|
KERNEL4x4_M1
|
|
KERNEL4x4_M2
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L4_M4_22
|
|
|
|
|
|
.Lctrmm_kernel_L4_M4_22a:
|
|
|
|
KERNEL4x4_M1
|
|
KERNEL4x4_E
|
|
|
|
b .Lctrmm_kernel_L4_M4_44
|
|
|
|
.Lctrmm_kernel_L4_M4_32:
|
|
|
|
tst counterL, #1
|
|
ble .Lctrmm_kernel_L4_M4_40
|
|
|
|
KERNEL4x4_I
|
|
KERNEL4x4_E
|
|
|
|
b .Lctrmm_kernel_L4_M4_44
|
|
|
|
|
|
.Lctrmm_kernel_L4_M4_40:
|
|
|
|
INIT4x4
|
|
|
|
.Lctrmm_kernel_L4_M4_44:
|
|
|
|
ands counterL , tempK, #1
|
|
ble .Lctrmm_kernel_L4_M4_100
|
|
|
|
.Lctrmm_kernel_L4_M4_46:
|
|
KERNEL4x4_SUB
|
|
|
|
.Lctrmm_kernel_L4_M4_100:
|
|
|
|
SAVE4x4
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#if defined(LEFT)
|
|
sub tempK, tempK, #4
|
|
#else
|
|
sub tempK, tempK, #4
|
|
#endif
|
|
lsl temp, tempK, #5
|
|
add pA, pA, temp
|
|
add pB, pB, temp
|
|
#endif
|
|
#if defined(LEFT)
|
|
add tempOffset, tempOffset, #4
|
|
#endif
|
|
|
|
.Lctrmm_kernel_L4_M4_END:
|
|
subs counterI, counterI, #1
|
|
bne .Lctrmm_kernel_L4_M4_20
|
|
|
|
.Lctrmm_kernel_L4_M2_BEGIN:
|
|
|
|
mov counterI, origM
|
|
tst counterI , #3
|
|
ble .Lctrmm_kernel_L4_END
|
|
|
|
tst counterI, #2 // counterI = counterI / 2
|
|
ble .Lctrmm_kernel_L4_M1_BEGIN
|
|
|
|
.Lctrmm_kernel_L4_M2_20:
|
|
|
|
INIT2x4
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
mov pB, origPB
|
|
#else
|
|
mov pB, origPB
|
|
lsl temp, tempOffset, #4
|
|
add pA, pA, temp
|
|
lsl temp, tempOffset, #5
|
|
add pB, pB, temp
|
|
#endif
|
|
|
|
#if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#elif defined(LEFT)
|
|
add tempK, tempOffset, #2
|
|
#else
|
|
add tempK, tempOffset, #4
|
|
#endif
|
|
|
|
asr counterL , tempK, #3 // counterL = counterL / 8
|
|
cmp counterL , #0
|
|
ble .Lctrmm_kernel_L4_M2_40
|
|
|
|
.Lctrmm_kernel_L4_M2_22:
|
|
|
|
KERNEL2x4_SUB
|
|
KERNEL2x4_SUB
|
|
KERNEL2x4_SUB
|
|
KERNEL2x4_SUB
|
|
|
|
KERNEL2x4_SUB
|
|
KERNEL2x4_SUB
|
|
KERNEL2x4_SUB
|
|
KERNEL2x4_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L4_M2_22
|
|
|
|
|
|
.Lctrmm_kernel_L4_M2_40:
|
|
|
|
ands counterL , tempK, #7 // counterL = counterL % 8
|
|
ble .Lctrmm_kernel_L4_M2_100
|
|
|
|
.Lctrmm_kernel_L4_M2_42:
|
|
|
|
KERNEL2x4_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L4_M2_42
|
|
|
|
.Lctrmm_kernel_L4_M2_100:
|
|
|
|
SAVE2x4
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#if defined(LEFT)
|
|
sub tempK, tempK, #2
|
|
#else
|
|
sub tempK, tempK, #4
|
|
#endif
|
|
lsl temp, tempK, #4
|
|
add pA, pA, temp
|
|
lsl temp, tempK, #5
|
|
add pB, pB, temp
|
|
#endif
|
|
#if defined(LEFT)
|
|
add tempOffset, tempOffset, #2
|
|
#endif
|
|
|
|
.Lctrmm_kernel_L4_M2_END:
|
|
|
|
|
|
.Lctrmm_kernel_L4_M1_BEGIN:
|
|
|
|
tst counterI, #1 // counterI = counterI % 2
|
|
ble .Lctrmm_kernel_L4_END
|
|
|
|
.Lctrmm_kernel_L4_M1_20:
|
|
|
|
INIT1x4
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
mov pB, origPB
|
|
#else
|
|
mov pB, origPB
|
|
lsl temp, tempOffset, #5
|
|
add pB, pB, temp
|
|
lsl temp, tempOffset, #3
|
|
add pA, pA, temp
|
|
#endif
|
|
|
|
#if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#elif defined(LEFT)
|
|
add tempK, tempOffset, #1
|
|
#else
|
|
add tempK, tempOffset, #4
|
|
#endif
|
|
|
|
asr counterL , tempK, #3 // counterL = counterL / 8
|
|
cmp counterL , #0
|
|
ble .Lctrmm_kernel_L4_M1_40
|
|
|
|
.Lctrmm_kernel_L4_M1_22:
|
|
KERNEL1x4_SUB
|
|
KERNEL1x4_SUB
|
|
KERNEL1x4_SUB
|
|
KERNEL1x4_SUB
|
|
|
|
KERNEL1x4_SUB
|
|
KERNEL1x4_SUB
|
|
KERNEL1x4_SUB
|
|
KERNEL1x4_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L4_M1_22
|
|
|
|
|
|
.Lctrmm_kernel_L4_M1_40:
|
|
|
|
ands counterL , tempK, #7 // counterL = counterL % 8
|
|
ble .Lctrmm_kernel_L4_M1_100
|
|
|
|
.Lctrmm_kernel_L4_M1_42:
|
|
|
|
KERNEL1x4_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L4_M1_42
|
|
|
|
.Lctrmm_kernel_L4_M1_100:
|
|
|
|
SAVE1x4
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#if defined(LEFT)
|
|
sub tempK, tempK, #1
|
|
#else
|
|
sub tempK, tempK, #4
|
|
#endif
|
|
lsl temp, tempK, #3
|
|
add pA, pA, temp
|
|
lsl temp, tempK, #5
|
|
add pB, pB, temp
|
|
#endif
|
|
#if defined(LEFT)
|
|
add tempOffset, tempOffset, #1
|
|
#endif
|
|
|
|
.Lctrmm_kernel_L4_END:
|
|
|
|
lsl temp, origK, #5
|
|
add origPB, origPB, temp // B = B + K * 4 * 8
|
|
|
|
#if !defined(LEFT)
|
|
add tempOffset, tempOffset, #4
|
|
#endif
|
|
|
|
subs counterJ, counterJ , #1 // j--
|
|
bgt .Lctrmm_kernel_L4_BEGIN
|
|
|
|
|
|
/******************************************************************************/
|
|
|
|
.Lctrmm_kernel_L2_BEGIN: // less than 2 left in N direction
|
|
|
|
mov counterJ , origN
|
|
tst counterJ , #3
|
|
ble .Lctrmm_kernel_L999 // error, N was less than 4?
|
|
|
|
tst counterJ , #2
|
|
ble .Lctrmm_kernel_L1_BEGIN
|
|
|
|
mov pCRow0, pC // pCRow0 = pC
|
|
|
|
add pC,pC,LDC, lsl #1
|
|
|
|
#if defined(LEFT)
|
|
mov tempOffset, offset
|
|
#endif
|
|
|
|
mov pA, origPA // pA = A
|
|
|
|
.Lctrmm_kernel_L2_M4_BEGIN:
|
|
|
|
mov counterI, origM
|
|
asr counterI, counterI, #2 // counterI = counterI / 4
|
|
cmp counterI,#0
|
|
ble .Lctrmm_kernel_L2_M2_BEGIN
|
|
|
|
.Lctrmm_kernel_L2_M4_20:
|
|
|
|
INIT4x2
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
mov pB, origPB
|
|
#else
|
|
mov pB, origPB
|
|
lsl temp, tempOffset, #4
|
|
add pB, pB, temp
|
|
lsl temp, tempOffset, #5
|
|
add pA, pA, temp
|
|
#endif
|
|
|
|
#if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#elif defined(LEFT)
|
|
add tempK, tempOffset, #4
|
|
#else
|
|
add tempK, tempOffset, #2
|
|
#endif
|
|
|
|
asr counterL , tempK, #3 // counterL = counterL / 8
|
|
cmp counterL,#0
|
|
ble .Lctrmm_kernel_L2_M4_40
|
|
.align 5
|
|
|
|
.Lctrmm_kernel_L2_M4_22:
|
|
KERNEL4x2_SUB
|
|
KERNEL4x2_SUB
|
|
KERNEL4x2_SUB
|
|
KERNEL4x2_SUB
|
|
|
|
KERNEL4x2_SUB
|
|
KERNEL4x2_SUB
|
|
KERNEL4x2_SUB
|
|
KERNEL4x2_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L2_M4_22
|
|
|
|
|
|
.Lctrmm_kernel_L2_M4_40:
|
|
|
|
ands counterL , tempK, #7 // counterL = counterL % 8
|
|
ble .Lctrmm_kernel_L2_M4_100
|
|
|
|
.Lctrmm_kernel_L2_M4_42:
|
|
|
|
KERNEL4x2_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L2_M4_42
|
|
|
|
.Lctrmm_kernel_L2_M4_100:
|
|
|
|
SAVE4x2
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#if defined(LEFT)
|
|
sub tempK, tempK, #4
|
|
#else
|
|
sub tempK, tempK, #2
|
|
#endif
|
|
lsl temp, tempK, #5
|
|
add pA, pA, temp
|
|
lsl temp, tempK, #4
|
|
add pB, pB, temp
|
|
#endif
|
|
#if defined(LEFT)
|
|
add tempOffset, tempOffset, #4
|
|
#endif
|
|
|
|
.Lctrmm_kernel_L2_M4_END:
|
|
|
|
subs counterI, counterI, #1
|
|
bgt .Lctrmm_kernel_L2_M4_20
|
|
|
|
|
|
.Lctrmm_kernel_L2_M2_BEGIN:
|
|
|
|
mov counterI, origM
|
|
tst counterI , #3
|
|
ble .Lctrmm_kernel_L2_END
|
|
|
|
tst counterI, #2 // counterI = counterI / 2
|
|
ble .Lctrmm_kernel_L2_M1_BEGIN
|
|
|
|
.Lctrmm_kernel_L2_M2_20:
|
|
|
|
INIT2x2
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
mov pB, origPB
|
|
#else
|
|
mov pB, origPB
|
|
lsl temp, tempOffset, #4
|
|
add pB, pB, temp
|
|
lsl temp, tempOffset, #4
|
|
add pA, pA, temp
|
|
#endif
|
|
|
|
#if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#elif defined(LEFT)
|
|
add tempK, tempOffset, #2
|
|
#else
|
|
add tempK, tempOffset, #2
|
|
#endif
|
|
|
|
asr counterL , tempK, #3 // counterL = counterL / 8
|
|
cmp counterL,#0
|
|
ble .Lctrmm_kernel_L2_M2_40
|
|
|
|
.Lctrmm_kernel_L2_M2_22:
|
|
|
|
KERNEL2x2_SUB
|
|
KERNEL2x2_SUB
|
|
KERNEL2x2_SUB
|
|
KERNEL2x2_SUB
|
|
|
|
KERNEL2x2_SUB
|
|
KERNEL2x2_SUB
|
|
KERNEL2x2_SUB
|
|
KERNEL2x2_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L2_M2_22
|
|
|
|
|
|
.Lctrmm_kernel_L2_M2_40:
|
|
|
|
ands counterL , tempK, #7 // counterL = counterL % 8
|
|
ble .Lctrmm_kernel_L2_M2_100
|
|
|
|
.Lctrmm_kernel_L2_M2_42:
|
|
|
|
KERNEL2x2_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L2_M2_42
|
|
|
|
.Lctrmm_kernel_L2_M2_100:
|
|
|
|
SAVE2x2
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#if defined(LEFT)
|
|
sub tempK, tempK, #2
|
|
#else
|
|
sub tempK, tempK, #2
|
|
#endif
|
|
lsl temp, tempK, #4
|
|
add pA, pA, temp
|
|
lsl temp, tempK, #4
|
|
add pB, pB, temp
|
|
#endif
|
|
#if defined(LEFT)
|
|
add tempOffset, tempOffset, #2
|
|
#endif
|
|
|
|
.Lctrmm_kernel_L2_M2_END:
|
|
|
|
|
|
.Lctrmm_kernel_L2_M1_BEGIN:
|
|
|
|
tst counterI, #1 // counterI = counterI % 2
|
|
ble .Lctrmm_kernel_L2_END
|
|
|
|
.Lctrmm_kernel_L2_M1_20:
|
|
|
|
INIT1x2
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
mov pB, origPB
|
|
#else
|
|
mov pB, origPB
|
|
lsl temp, tempOffset, #4
|
|
add pB, pB, temp
|
|
lsl temp, tempOffset, #3
|
|
add pA, pA, temp
|
|
#endif
|
|
|
|
#if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#elif defined(LEFT)
|
|
add tempK, tempOffset, #1
|
|
#else
|
|
add tempK, tempOffset, #2
|
|
#endif
|
|
|
|
asr counterL , tempK, #3 // counterL = counterL / 8
|
|
cmp counterL, #0
|
|
ble .Lctrmm_kernel_L2_M1_40
|
|
|
|
.Lctrmm_kernel_L2_M1_22:
|
|
KERNEL1x2_SUB
|
|
KERNEL1x2_SUB
|
|
KERNEL1x2_SUB
|
|
KERNEL1x2_SUB
|
|
|
|
KERNEL1x2_SUB
|
|
KERNEL1x2_SUB
|
|
KERNEL1x2_SUB
|
|
KERNEL1x2_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L2_M1_22
|
|
|
|
|
|
.Lctrmm_kernel_L2_M1_40:
|
|
|
|
ands counterL , tempK, #7 // counterL = counterL % 8
|
|
ble .Lctrmm_kernel_L2_M1_100
|
|
|
|
.Lctrmm_kernel_L2_M1_42:
|
|
|
|
KERNEL1x2_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L2_M1_42
|
|
|
|
.Lctrmm_kernel_L2_M1_100:
|
|
|
|
SAVE1x2
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#if defined(LEFT)
|
|
sub tempK, tempK, #1
|
|
#else
|
|
sub tempK, tempK, #2
|
|
#endif
|
|
lsl temp, tempK, #3
|
|
add pA, pA, temp
|
|
lsl temp, tempK, #4
|
|
add pB, pB, temp
|
|
#endif
|
|
#if defined(LEFT)
|
|
add tempOffset, tempOffset, #1
|
|
#endif
|
|
|
|
.Lctrmm_kernel_L2_END:
|
|
#if !defined(LEFT)
|
|
add tempOffset, tempOffset, #2
|
|
#endif
|
|
add origPB, origPB, origK, lsl #4 // B = B + K * 2 * 8
|
|
|
|
/******************************************************************************/
|
|
|
|
.Lctrmm_kernel_L1_BEGIN:
|
|
|
|
mov counterJ , origN
|
|
tst counterJ , #1
|
|
ble .Lctrmm_kernel_L999 // done
|
|
|
|
|
|
mov pCRow0, pC // pCRow0 = C
|
|
add pC , pC , LDC // Update pC to point to next
|
|
|
|
#if defined(LEFT)
|
|
mov tempOffset, offset
|
|
#endif
|
|
|
|
mov pA, origPA // pA = A
|
|
|
|
.Lctrmm_kernel_L1_M4_BEGIN:
|
|
|
|
mov counterI, origM
|
|
asr counterI, counterI, #2 // counterI = counterI / 4
|
|
cmp counterI, #0
|
|
ble .Lctrmm_kernel_L1_M2_BEGIN
|
|
|
|
.Lctrmm_kernel_L1_M4_20:
|
|
|
|
INIT4x1
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
mov pB, origPB
|
|
#else
|
|
mov pB, origPB
|
|
lsl temp, tempOffset, #3
|
|
add pB, pB, temp
|
|
lsl temp, tempOffset, #5
|
|
add pA, pA, temp
|
|
#endif
|
|
|
|
#if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#elif defined(LEFT)
|
|
add tempK, tempOffset, #4
|
|
#else
|
|
add tempK, tempOffset, #1
|
|
#endif
|
|
|
|
asr counterL , tempK, #3 // counterL = counterL / 8
|
|
cmp counterL , #0
|
|
ble .Lctrmm_kernel_L1_M4_40
|
|
.align 5
|
|
|
|
.Lctrmm_kernel_L1_M4_22:
|
|
KERNEL4x1_SUB
|
|
KERNEL4x1_SUB
|
|
KERNEL4x1_SUB
|
|
KERNEL4x1_SUB
|
|
|
|
KERNEL4x1_SUB
|
|
KERNEL4x1_SUB
|
|
KERNEL4x1_SUB
|
|
KERNEL4x1_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L1_M4_22
|
|
|
|
|
|
.Lctrmm_kernel_L1_M4_40:
|
|
|
|
ands counterL , tempK, #7 // counterL = counterL % 8
|
|
ble .Lctrmm_kernel_L1_M4_100
|
|
|
|
.Lctrmm_kernel_L1_M4_42:
|
|
|
|
KERNEL4x1_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L1_M4_42
|
|
|
|
.Lctrmm_kernel_L1_M4_100:
|
|
|
|
SAVE4x1
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#if defined(LEFT)
|
|
sub tempK, tempK, #4
|
|
#else
|
|
sub tempK, tempK, #1
|
|
#endif
|
|
lsl temp, tempK, #5
|
|
add pA, pA, temp
|
|
lsl temp, tempK, #3
|
|
add pB, pB, temp
|
|
#endif
|
|
#if defined(LEFT)
|
|
add tempOffset, tempOffset, #4
|
|
#endif
|
|
|
|
.Lctrmm_kernel_L1_M4_END:
|
|
|
|
subs counterI, counterI, #1
|
|
bgt .Lctrmm_kernel_L1_M4_20
|
|
|
|
|
|
.Lctrmm_kernel_L1_M2_BEGIN:
|
|
|
|
mov counterI, origM
|
|
tst counterI , #3
|
|
ble .Lctrmm_kernel_L1_END
|
|
|
|
tst counterI, #2 // counterI = counterI / 2
|
|
ble .Lctrmm_kernel_L1_M1_BEGIN
|
|
|
|
.Lctrmm_kernel_L1_M2_20:
|
|
|
|
INIT2x1
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
mov pB, origPB
|
|
#else
|
|
mov pB, origPB
|
|
lsl temp, tempOffset, #3
|
|
add pB, pB, temp
|
|
lsl temp, tempOffset, #4
|
|
add pA, pA, temp
|
|
#endif
|
|
|
|
#if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#elif defined(LEFT)
|
|
add tempK, tempOffset, #2
|
|
#else
|
|
add tempK, tempOffset, #1
|
|
#endif
|
|
|
|
asr counterL , tempK, #3 // counterL = counterL / 8
|
|
cmp counterL , #0
|
|
ble .Lctrmm_kernel_L1_M2_40
|
|
|
|
.Lctrmm_kernel_L1_M2_22:
|
|
|
|
KERNEL2x1_SUB
|
|
KERNEL2x1_SUB
|
|
KERNEL2x1_SUB
|
|
KERNEL2x1_SUB
|
|
|
|
KERNEL2x1_SUB
|
|
KERNEL2x1_SUB
|
|
KERNEL2x1_SUB
|
|
KERNEL2x1_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L1_M2_22
|
|
|
|
|
|
.Lctrmm_kernel_L1_M2_40:
|
|
|
|
ands counterL , tempK, #7 // counterL = counterL % 8
|
|
ble .Lctrmm_kernel_L1_M2_100
|
|
|
|
.Lctrmm_kernel_L1_M2_42:
|
|
|
|
KERNEL2x1_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L1_M2_42
|
|
|
|
.Lctrmm_kernel_L1_M2_100:
|
|
|
|
SAVE2x1
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#if defined(LEFT)
|
|
sub tempK, tempK, #2
|
|
#else
|
|
sub tempK, tempK, #1
|
|
#endif
|
|
lsl temp, tempK, #4
|
|
add pA, pA, temp
|
|
lsl temp, tempK, #3
|
|
add pB, pB, temp
|
|
#endif
|
|
#if defined(LEFT)
|
|
add tempOffset, tempOffset, #2
|
|
#endif
|
|
|
|
.Lctrmm_kernel_L1_M2_END:
|
|
|
|
|
|
.Lctrmm_kernel_L1_M1_BEGIN:
|
|
|
|
tst counterI, #1 // counterI = counterI % 2
|
|
ble .Lctrmm_kernel_L1_END
|
|
|
|
.Lctrmm_kernel_L1_M1_20:
|
|
|
|
INIT1x1
|
|
|
|
#if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
|
|
mov pB, origPB
|
|
#else
|
|
mov pB, origPB
|
|
lsl temp, tempOffset, #3
|
|
add pB, pB, temp
|
|
lsl temp, tempOffset, #3
|
|
add pA, pA, temp
|
|
#endif
|
|
|
|
#if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
|
|
sub tempK, origK, tempOffset
|
|
#elif defined(LEFT)
|
|
add tempK, tempOffset, #1
|
|
#else
|
|
add tempK, tempOffset, #1
|
|
#endif
|
|
|
|
asr counterL , tempK, #3 // counterL = counterL / 8
|
|
cmp counterL , #0
|
|
ble .Lctrmm_kernel_L1_M1_40
|
|
|
|
.Lctrmm_kernel_L1_M1_22:
|
|
KERNEL1x1_SUB
|
|
KERNEL1x1_SUB
|
|
KERNEL1x1_SUB
|
|
KERNEL1x1_SUB
|
|
|
|
KERNEL1x1_SUB
|
|
KERNEL1x1_SUB
|
|
KERNEL1x1_SUB
|
|
KERNEL1x1_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L1_M1_22
|
|
|
|
|
|
.Lctrmm_kernel_L1_M1_40:
|
|
|
|
ands counterL , tempK, #7 // counterL = counterL % 8
|
|
ble .Lctrmm_kernel_L1_M1_100
|
|
|
|
.Lctrmm_kernel_L1_M1_42:
|
|
|
|
KERNEL1x1_SUB
|
|
|
|
subs counterL, counterL, #1
|
|
bgt .Lctrmm_kernel_L1_M1_42
|
|
|
|
.Lctrmm_kernel_L1_M1_100:
|
|
|
|
SAVE1x1
|
|
|
|
|
|
.Lctrmm_kernel_L1_END:
|
|
|
|
|
|
.Lctrmm_kernel_L999:
|
|
mov x0, #0 // set return value
|
|
ldp d8, d9, [sp, #(0 * 16)]
|
|
ldp d10, d11, [sp, #(1 * 16)]
|
|
ldp d12, d13, [sp, #(2 * 16)]
|
|
ldp d14, d15, [sp, #(3 * 16)]
|
|
ldp d16, d17, [sp, #(4 * 16)]
|
|
ldp x18, x19, [sp, #(5 * 16)]
|
|
ldp x20, x21, [sp, #(6 * 16)]
|
|
ldp x22, x23, [sp, #(7 * 16)]
|
|
ldp x24, x25, [sp, #(8 * 16)]
|
|
ldp x26, x27, [sp, #(9 * 16)]
|
|
ldr x28, [sp, #(10 * 16)]
|
|
add sp, sp, #(11*16)
|
|
ret
|
|
|
|
EPILOGUE
|
|
|