| @@ -90,19 +90,17 @@ SGEMMKERNEL = sgemm_kernel_4x4_vfpv3.S | |||
| SGEMMINCOPY = | |||
| SGEMMITCOPY = | |||
| SGEMMONCOPY = sgemm_ncopy_4_vfpv3.S | |||
| SGEMMOTCOPY = ../generic/gemm_tcopy_4.c | |||
| SGEMMOTCOPY = sgemm_tcopy_4_vfpv3.S | |||
| SGEMMINCOPYOBJ = | |||
| SGEMMITCOPYOBJ = | |||
| SGEMMONCOPYOBJ = sgemm_oncopy.o | |||
| SGEMMOTCOPYOBJ = sgemm_otcopy.o | |||
| #DGEMMKERNEL = ../generic/gemmkernel_2x2.c | |||
| #DGEMMKERNEL = dgemm_kernel_4x2_vfpv2.S | |||
| DGEMMKERNEL = dgemm_kernel_4x4_vfpv3.S | |||
| DGEMMINCOPY = | |||
| DGEMMITCOPY = | |||
| DGEMMONCOPY = dgemm_ncopy_4_vfpv3.S | |||
| DGEMMOTCOPY = ../generic/gemm_tcopy_4.c | |||
| DGEMMOTCOPY = dgemm_tcopy_4_vfpv3.S | |||
| DGEMMINCOPYOBJ = | |||
| DGEMMITCOPYOBJ = | |||
| DGEMMONCOPYOBJ = dgemm_oncopy.o | |||
| @@ -77,7 +77,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |||
| #define B [fp, #4 ] | |||
| #define C [fp, #8 ] | |||
| #define OLDdgemm_kernel_LDC [fp, #12 ] | |||
| #define OLD_LDC [fp, #12 ] | |||
| #define I r0 | |||
| #define J r1 | |||
| @@ -883,7 +883,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |||
| sub r3, fp, #128 | |||
| vstm r3, { d8 - d15} // store floating point registers | |||
| ldr r3, OLDdgemm_kernel_LDC | |||
| ldr r3, OLD_LDC | |||
| lsl r3, r3, #3 // ldc = ldc * 8 | |||
| str r3, LDC | |||
| @@ -0,0 +1,408 @@ | |||
| /*************************************************************************** | |||
| Copyright (c) 2013, 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. | |||
| *****************************************************************************/ | |||
| /************************************************************************************** | |||
| * 2013/11/06 Saar | |||
| * BLASTEST : OK | |||
| * CTEST : OK | |||
| * TEST : OK | |||
| * | |||
| **************************************************************************************/ | |||
| #define ASSEMBLER | |||
| #include "common.h" | |||
| #define STACKSIZE 256 | |||
| #define OLD_M r0 | |||
| #define OLD_N r1 | |||
| #define OLD_A r2 | |||
| #define OLD_LDA r3 | |||
| /****************************************************** | |||
| * [fp, #-128] - [fp, #-64] is reserved | |||
| * for store and restore of floating point | |||
| * registers | |||
| *******************************************************/ | |||
| #define B [fp, #4 ] | |||
| #define A [fp, #-248 ] | |||
| #define M r0 | |||
| #define N r1 | |||
| #define M4 r2 | |||
| #define LDA r5 | |||
| #define AO1 r6 | |||
| #define BO1 r7 | |||
| #define BO2 r8 | |||
| #define BO3 r9 | |||
| #define I r4 | |||
| #define J r12 | |||
| #define A_PRE 256 | |||
| /************************************************************************************** | |||
| * Macro definitions | |||
| **************************************************************************************/ | |||
| .macro COPY4x4 | |||
| pld [ AO1, #A_PRE ] | |||
| fldmiad AO1, { d0 - d3 } | |||
| add r3, AO1, LDA | |||
| pld [ r3, #A_PRE ] | |||
| fldmiad r3, { d4 - d7 } | |||
| add r3, r3, LDA | |||
| pld [ r3, #A_PRE ] | |||
| fldmiad r3, { d8 - d11 } | |||
| add r3, r3, LDA | |||
| pld [ r3, #A_PRE ] | |||
| fldmiad r3, { d12 - d15 } | |||
| fstmiad BO1, { d0 - d15 } | |||
| add AO1, AO1, #32 | |||
| add BO1, BO1, M4 | |||
| .endm | |||
| .macro COPY2x4 | |||
| fldmiad AO1, { d0 - d1 } | |||
| add r3, AO1, LDA | |||
| fldmiad r3, { d2 - d3 } | |||
| add r3, r3, LDA | |||
| fldmiad r3, { d4 - d5 } | |||
| add r3, r3, LDA | |||
| fldmiad r3, { d6 - d7 } | |||
| fstmiad BO2, { d0 - d7 } | |||
| add AO1, AO1, #16 | |||
| add BO2, BO2, #64 | |||
| .endm | |||
| .macro COPY1x4 | |||
| fldmiad AO1, { d0 } | |||
| add r3, AO1, LDA | |||
| fldmiad r3, { d1 } | |||
| add r3, r3, LDA | |||
| fldmiad r3, { d2 } | |||
| add r3, r3, LDA | |||
| fldmiad r3, { d3 } | |||
| fstmiad BO3, { d0 - d3 } | |||
| add AO1, AO1, #8 | |||
| add BO3, BO3, #32 | |||
| .endm | |||
| /*************************************************************************************************************************/ | |||
| .macro COPY4x2 | |||
| pld [ AO1, #A_PRE ] | |||
| fldmiad AO1, { d0 - d3 } | |||
| add r3, AO1, LDA | |||
| pld [ r3, #A_PRE ] | |||
| fldmiad r3, { d4 - d7 } | |||
| fstmiad BO1, { d0 - d7 } | |||
| add AO1, AO1, #32 | |||
| add BO1, BO1, M4 | |||
| .endm | |||
| .macro COPY2x2 | |||
| fldmiad AO1, { d0 - d1 } | |||
| add r3, AO1, LDA | |||
| fldmiad r3, { d2 - d3 } | |||
| fstmiad BO2, { d0 - d3 } | |||
| add AO1, AO1, #16 | |||
| add BO2, BO2, #32 | |||
| .endm | |||
| .macro COPY1x2 | |||
| fldmiad AO1, { d0 } | |||
| add r3, AO1, LDA | |||
| fldmiad r3, { d1 } | |||
| fstmiad BO3, { d0 - d1 } | |||
| add AO1, AO1, #8 | |||
| add BO3, BO3, #16 | |||
| .endm | |||
| /*************************************************************************************************************************/ | |||
| .macro COPY4x1 | |||
| pld [ AO1, #A_PRE ] | |||
| fldmiad AO1, { d0 - d3 } | |||
| fstmiad BO1, { d0 - d3 } | |||
| add AO1, AO1, #32 | |||
| add BO1, BO1, M4 | |||
| .endm | |||
| .macro COPY2x1 | |||
| fldmiad AO1, { d0 - d1 } | |||
| fstmiad BO2, { d0 - d1 } | |||
| add AO1, AO1, #16 | |||
| add BO2, BO2, #16 | |||
| .endm | |||
| .macro COPY1x1 | |||
| fldmiad AO1, { d0 } | |||
| fstmiad BO3, { d0 } | |||
| add AO1, AO1, #8 | |||
| add BO3, BO3, #8 | |||
| .endm | |||
| /************************************************************************************** | |||
| * End of macro definitions | |||
| **************************************************************************************/ | |||
| PROLOGUE | |||
| .align 5 | |||
| push {r4 - r9, fp} | |||
| add fp, sp, #24 | |||
| sub sp, sp, #STACKSIZE // reserve stack | |||
| str OLD_A, A // store A | |||
| lsl LDA, OLD_LDA, #3 // lda = lda * SIZE | |||
| sub r4, fp, #128 | |||
| vstm r4, { d8 - d15} // store floating point registers | |||
| lsl r4 , M, #3 // M * SIZE | |||
| ldr r3, B | |||
| and BO2 , N , #-4 | |||
| and BO3 , N , #-2 | |||
| mul BO2, BO2, r4 | |||
| mul BO3, BO3, r4 | |||
| add BO2 , BO2, r3 | |||
| add BO3 , BO3, r3 | |||
| lsl M4, M, #5 // M4 = M * 4 * SIZE | |||
| dgemm_tcopy_L4_BEGIN: | |||
| asrs J, M, #2 // J = N / 4 | |||
| ble dgemm_tcopy_L2_BEGIN | |||
| dgemm_tcopy_L4_M4_BEGIN: | |||
| ldr AO1, A // AO1 = A | |||
| lsl r3, LDA, #2 // r3 = 4 * LDA | |||
| add r3, r3 , AO1 // A = A + 4 * LDA | |||
| str r3, A // store A | |||
| ldr BO1, B | |||
| add r3, BO1, #128 // B = B + 16 * SIZE | |||
| str r3, B | |||
| asrs I, N, #2 // I = M / 4 | |||
| ble dgemm_tcopy_L4_M4_40 | |||
| dgemm_tcopy_L4_M4_20: | |||
| COPY4x4 | |||
| subs I , I , #1 | |||
| bne dgemm_tcopy_L4_M4_20 | |||
| dgemm_tcopy_L4_M4_40: | |||
| tst N , #2 | |||
| ble dgemm_tcopy_L4_M4_60 | |||
| COPY2x4 | |||
| dgemm_tcopy_L4_M4_60: | |||
| tst N, #1 | |||
| ble dgemm_tcopy_L4_M4_END | |||
| COPY1x4 | |||
| dgemm_tcopy_L4_M4_END: | |||
| subs J , J, #1 // j-- | |||
| bne dgemm_tcopy_L4_M4_BEGIN | |||
| /*********************************************************************************************/ | |||
| dgemm_tcopy_L2_BEGIN: | |||
| tst M, #3 | |||
| ble dgemm_tcopy_L999 | |||
| tst M, #2 | |||
| ble dgemm_tcopy_L1_BEGIN | |||
| dgemm_tcopy_L2_M4_BEGIN: | |||
| ldr AO1, A // AO1 = A | |||
| lsl r3, LDA, #1 // r3 = 2 * LDA | |||
| add r3, r3 , AO1 // A = A + 2 * LDA | |||
| str r3, A // store A | |||
| ldr BO1, B | |||
| add r3, BO1, #64 // B = B + 8 * SIZE | |||
| str r3, B | |||
| asrs I, N, #2 // I = M / 4 | |||
| ble dgemm_tcopy_L2_M4_40 | |||
| dgemm_tcopy_L2_M4_20: | |||
| COPY4x2 | |||
| subs I , I , #1 | |||
| bne dgemm_tcopy_L2_M4_20 | |||
| dgemm_tcopy_L2_M4_40: | |||
| tst N , #2 | |||
| ble dgemm_tcopy_L2_M4_60 | |||
| COPY2x2 | |||
| dgemm_tcopy_L2_M4_60: | |||
| tst N , #1 | |||
| ble dgemm_tcopy_L2_M4_END | |||
| COPY1x2 | |||
| dgemm_tcopy_L2_M4_END: | |||
| /*********************************************************************************************/ | |||
| dgemm_tcopy_L1_BEGIN: | |||
| tst M, #1 | |||
| ble dgemm_tcopy_L999 | |||
| dgemm_tcopy_L1_M4_BEGIN: | |||
| ldr AO1, A // AO1 = A | |||
| add r3, LDA , AO1 // A = A + 1 * LDA | |||
| str r3, A // store A | |||
| ldr BO1, B | |||
| add r3, BO1, #32 // B = B + 4 * SIZE | |||
| str r3, B | |||
| asrs I, N, #2 // I = M / 4 | |||
| ble dgemm_tcopy_L1_M4_40 | |||
| dgemm_tcopy_L1_M4_20: | |||
| COPY4x1 | |||
| subs I , I , #1 | |||
| bne dgemm_tcopy_L1_M4_20 | |||
| dgemm_tcopy_L1_M4_40: | |||
| tst N , #2 | |||
| ble dgemm_tcopy_L1_M4_60 | |||
| COPY2x1 | |||
| dgemm_tcopy_L1_M4_60: | |||
| tst N , #1 | |||
| ble dgemm_tcopy_L1_M4_END | |||
| COPY1x1 | |||
| dgemm_tcopy_L1_M4_END: | |||
| dgemm_tcopy_L999: | |||
| sub r3, fp, #128 | |||
| vldm r3, { d8 - d15} // restore floating point registers | |||
| mov r0, #0 // set return value | |||
| sub sp, fp, #24 | |||
| pop {r4 - r9, fp} | |||
| bx lr | |||
| EPILOGUE | |||
| @@ -0,0 +1,430 @@ | |||
| /*************************************************************************** | |||
| Copyright (c) 2013, 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. | |||
| *****************************************************************************/ | |||
| /************************************************************************************** | |||
| * 2013/11/06 Saar | |||
| * BLASTEST : OK | |||
| * CTEST : OK | |||
| * TEST : OK | |||
| * | |||
| **************************************************************************************/ | |||
| #define ASSEMBLER | |||
| #include "common.h" | |||
| #define STACKSIZE 256 | |||
| #define OLD_M r0 | |||
| #define OLD_N r1 | |||
| #define OLD_A r2 | |||
| #define OLD_LDA r3 | |||
| /****************************************************** | |||
| * [fp, #-128] - [fp, #-64] is reserved | |||
| * for store and restore of floating point | |||
| * registers | |||
| *******************************************************/ | |||
| #define B [fp, #4 ] | |||
| #define A [fp, #-248 ] | |||
| #define M r0 | |||
| #define N r1 | |||
| #define M4 r2 | |||
| #define LDA r5 | |||
| #define AO1 r6 | |||
| #define BO1 r7 | |||
| #define BO2 r8 | |||
| #define BO3 r9 | |||
| #define I r4 | |||
| #define J r12 | |||
| #define A_PRE 256 | |||
| /************************************************************************************** | |||
| * Macro definitions | |||
| **************************************************************************************/ | |||
| .macro COPY4x4_1 | |||
| pld [ AO1, #A_PRE ] | |||
| fldmias AO1, { s0 - s3 } | |||
| add r3, AO1, LDA | |||
| pld [ r3, #A_PRE ] | |||
| fldmias r3, { s4 - s7 } | |||
| add r3, r3, LDA | |||
| pld [ r3, #A_PRE ] | |||
| fldmias r3, { s8 - s11 } | |||
| add r3, r3, LDA | |||
| pld [ r3, #A_PRE ] | |||
| fldmias r3, { s12 - s15 } | |||
| fstmias BO1, { s0 - s15 } | |||
| add AO1, AO1, #16 | |||
| add BO1, BO1, M4 | |||
| .endm | |||
| .macro COPY4x4_2 | |||
| fldmias AO1, { s0 - s3 } | |||
| add r3, AO1, LDA | |||
| fldmias r3, { s4 - s7 } | |||
| add r3, r3, LDA | |||
| fldmias r3, { s8 - s11 } | |||
| add r3, r3, LDA | |||
| fldmias r3, { s12 - s15 } | |||
| fstmias BO1, { s0 - s15 } | |||
| add AO1, AO1, #16 | |||
| add BO1, BO1, M4 | |||
| .endm | |||
| .macro COPY2x4 | |||
| fldmias AO1, { s0 - s1 } | |||
| add r3, AO1, LDA | |||
| fldmias r3, { s2 - s3 } | |||
| add r3, r3, LDA | |||
| fldmias r3, { s4 - s5 } | |||
| add r3, r3, LDA | |||
| fldmias r3, { s6 - s7 } | |||
| fstmias BO2, { s0 - s7 } | |||
| add AO1, AO1, #8 | |||
| add BO2, BO2, #32 | |||
| .endm | |||
| .macro COPY1x4 | |||
| fldmias AO1, { s0 } | |||
| add r3, AO1, LDA | |||
| fldmias r3, { s1 } | |||
| add r3, r3, LDA | |||
| fldmias r3, { s2 } | |||
| add r3, r3, LDA | |||
| fldmias r3, { s3 } | |||
| fstmias BO3, { s0 - s3 } | |||
| add AO1, AO1, #4 | |||
| add BO3, BO3, #16 | |||
| .endm | |||
| /*************************************************************************************************************************/ | |||
| .macro COPY4x2 | |||
| fldmias AO1, { s0 - s3 } | |||
| add r3, AO1, LDA | |||
| fldmias r3, { s4 - s7 } | |||
| fstmias BO1, { s0 - s7 } | |||
| add AO1, AO1, #16 | |||
| add BO1, BO1, M4 | |||
| .endm | |||
| .macro COPY2x2 | |||
| fldmias AO1, { s0 - s1 } | |||
| add r3, AO1, LDA | |||
| fldmias r3, { s2 - s3 } | |||
| fstmias BO2, { s0 - s3 } | |||
| add AO1, AO1, #8 | |||
| add BO2, BO2, #16 | |||
| .endm | |||
| .macro COPY1x2 | |||
| fldmias AO1, { s0 } | |||
| add r3, AO1, LDA | |||
| fldmias r3, { s1 } | |||
| fstmias BO3, { s0 - s1 } | |||
| add AO1, AO1, #4 | |||
| add BO3, BO3, #8 | |||
| .endm | |||
| /*************************************************************************************************************************/ | |||
| .macro COPY4x1 | |||
| fldmias AO1, { s0 - s3 } | |||
| fstmias BO1, { s0 - s3 } | |||
| add AO1, AO1, #16 | |||
| add BO1, BO1, M4 | |||
| .endm | |||
| .macro COPY2x1 | |||
| fldmias AO1, { s0 - s1 } | |||
| fstmias BO2, { s0 - s1 } | |||
| add AO1, AO1, #8 | |||
| add BO2, BO2, #8 | |||
| .endm | |||
| .macro COPY1x1 | |||
| fldmias AO1, { s0 } | |||
| fstmias BO3, { s0 } | |||
| add AO1, AO1, #4 | |||
| add BO3, BO3, #4 | |||
| .endm | |||
| /************************************************************************************** | |||
| * End of macro definitions | |||
| **************************************************************************************/ | |||
| PROLOGUE | |||
| .align 5 | |||
| push {r4 - r9, fp} | |||
| add fp, sp, #24 | |||
| sub sp, sp, #STACKSIZE // reserve stack | |||
| str OLD_A, A // store A | |||
| lsl LDA, OLD_LDA, #2 // lda = lda * SIZE | |||
| sub r4, fp, #128 | |||
| vstm r4, { s8 - s15} // store floating point registers | |||
| lsl r4 , M, #2 // M * SIZE | |||
| ldr r3, B | |||
| and BO2 , N , #-4 | |||
| and BO3 , N , #-2 | |||
| mul BO2, BO2, r4 | |||
| mul BO3, BO3, r4 | |||
| add BO2 , BO2, r3 | |||
| add BO3 , BO3, r3 | |||
| lsl M4, M, #4 // M4 = M * 4 * SIZE | |||
| sgemm_tcopy_L4_BEGIN: | |||
| asrs J, M, #2 // J = N / 4 | |||
| ble sgemm_tcopy_L2_BEGIN | |||
| sgemm_tcopy_L4_M4_BEGIN: | |||
| ldr AO1, A // AO1 = A | |||
| lsl r3, LDA, #2 // r3 = 4 * LDA | |||
| add r3, r3 , AO1 // A = A + 4 * LDA | |||
| str r3, A // store A | |||
| ldr BO1, B | |||
| add r3, BO1, #64 // B = B + 16 * SIZE | |||
| str r3, B | |||
| asrs I, N, #2 // I = M / 4 | |||
| ble sgemm_tcopy_L4_M4_40 | |||
| sgemm_tcopy_L4_M4_20: | |||
| COPY4x4_1 | |||
| subs I , I , #1 | |||
| ble sgemm_tcopy_L4_M4_40 | |||
| COPY4x4_2 | |||
| subs I , I , #1 | |||
| bne sgemm_tcopy_L4_M4_20 | |||
| sgemm_tcopy_L4_M4_40: | |||
| tst N , #2 | |||
| ble sgemm_tcopy_L4_M4_60 | |||
| COPY2x4 | |||
| sgemm_tcopy_L4_M4_60: | |||
| tst N, #1 | |||
| ble sgemm_tcopy_L4_M4_END | |||
| COPY1x4 | |||
| sgemm_tcopy_L4_M4_END: | |||
| subs J , J, #1 // j-- | |||
| bne sgemm_tcopy_L4_M4_BEGIN | |||
| /*********************************************************************************************/ | |||
| sgemm_tcopy_L2_BEGIN: | |||
| tst M, #3 | |||
| ble sgemm_tcopy_L999 | |||
| tst M, #2 | |||
| ble sgemm_tcopy_L1_BEGIN | |||
| sgemm_tcopy_L2_M4_BEGIN: | |||
| ldr AO1, A // AO1 = A | |||
| lsl r3, LDA, #1 // r3 = 2 * LDA | |||
| add r3, r3 , AO1 // A = A + 2 * LDA | |||
| str r3, A // store A | |||
| ldr BO1, B | |||
| add r3, BO1, #32 // B = B + 8 * SIZE | |||
| str r3, B | |||
| asrs I, N, #2 // I = M / 4 | |||
| ble sgemm_tcopy_L2_M4_40 | |||
| sgemm_tcopy_L2_M4_20: | |||
| COPY4x2 | |||
| subs I , I , #1 | |||
| bne sgemm_tcopy_L2_M4_20 | |||
| sgemm_tcopy_L2_M4_40: | |||
| tst N , #2 | |||
| ble sgemm_tcopy_L2_M4_60 | |||
| COPY2x2 | |||
| sgemm_tcopy_L2_M4_60: | |||
| tst N , #1 | |||
| ble sgemm_tcopy_L2_M4_END | |||
| COPY1x2 | |||
| sgemm_tcopy_L2_M4_END: | |||
| /*********************************************************************************************/ | |||
| sgemm_tcopy_L1_BEGIN: | |||
| tst M, #1 | |||
| ble sgemm_tcopy_L999 | |||
| sgemm_tcopy_L1_M4_BEGIN: | |||
| ldr AO1, A // AO1 = A | |||
| add r3, LDA , AO1 // A = A + 1 * LDA | |||
| str r3, A // store A | |||
| ldr BO1, B | |||
| add r3, BO1, #16 // B = B + 4 * SIZE | |||
| str r3, B | |||
| asrs I, N, #2 // I = M / 4 | |||
| ble sgemm_tcopy_L1_M4_40 | |||
| sgemm_tcopy_L1_M4_20: | |||
| COPY4x1 | |||
| subs I , I , #1 | |||
| bne sgemm_tcopy_L1_M4_20 | |||
| sgemm_tcopy_L1_M4_40: | |||
| tst N , #2 | |||
| ble sgemm_tcopy_L1_M4_60 | |||
| COPY2x1 | |||
| sgemm_tcopy_L1_M4_60: | |||
| tst N , #1 | |||
| ble sgemm_tcopy_L1_M4_END | |||
| COPY1x1 | |||
| sgemm_tcopy_L1_M4_END: | |||
| sgemm_tcopy_L999: | |||
| sub r3, fp, #128 | |||
| vldm r3, { s8 - s15} // restore floating point registers | |||
| mov r0, #0 // set return value | |||
| sub sp, fp, #24 | |||
| pop {r4 - r9, fp} | |||
| bx lr | |||
| EPILOGUE | |||