added cgemm_tcopy_2_vfpv3.S and zgemm_tcopy_2_vfpv3.S
authorwernsaar <wernsaar@googlemail.com>
Thu, 7 Nov 2013 16:15:50 +0000 (17:15 +0100)
committerwernsaar <wernsaar@googlemail.com>
Thu, 7 Nov 2013 16:15:50 +0000 (17:15 +0100)
kernel/arm/cgemm_tcopy_2_vfpv3.S [new file with mode: 0644]
kernel/arm/zgemm_tcopy_2_vfpv3.S [new file with mode: 0644]

diff --git a/kernel/arm/cgemm_tcopy_2_vfpv3.S b/kernel/arm/cgemm_tcopy_2_vfpv3.S
new file mode 100644 (file)
index 0000000..9036b99
--- /dev/null
@@ -0,0 +1,243 @@
+/***************************************************************************
+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/07 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 I      r4
+#define        J       r12
+
+#define A_PRE  256
+
+/**************************************************************************************
+* Macro definitions
+**************************************************************************************/
+.macro COPY2x2
+
+       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 COPY1x2
+
+       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 COPY2x1
+
+       fldmias AO1, { s0 - s3 }
+
+       fstmias BO1, { s0 - s3 }
+       add     AO1, AO1, #16
+       add     BO1, BO1, M4
+
+.endm
+
+.macro COPY1x1
+
+       fldmias AO1, { s0 - s1 }
+
+       fstmias BO2, { s0 - s1 }
+       add     AO1, AO1, #8
+       add     BO2, BO2, #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 * 2
+
+       sub     r4, fp, #128
+       vstm    r4, { s8 - s15}                                 // store floating point registers
+
+       lsl     r4 , M, #3                                      // M * SIZE * 2
+
+       ldr     r3, B
+
+       and     BO2 , N , #-2
+
+       mul     BO2, BO2, r4
+
+       add     BO2 , BO2, r3
+
+       lsl     M4, M, #4                                       // M4 = M * 2 * SIZE * 2
+
+cgemm_tcopy_L2_BEGIN:
+
+       asrs    J, M, #1                                        // J = N / 2
+       ble     cgemm_tcopy_L1_BEGIN
+
+cgemm_tcopy_L2_M2_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 + 4 * SIZE *2
+       str     r3, B
+
+       asrs    I, N, #1                                        // I = M / 2
+       ble     cgemm_tcopy_L2_M2_60
+
+cgemm_tcopy_L2_M2_40:
+
+       COPY2x2
+       subs I, I, #1
+       bne     cgemm_tcopy_L2_M2_40
+
+cgemm_tcopy_L2_M2_60:
+
+       tst     N , #1
+       ble     cgemm_tcopy_L2_M2_END
+
+       COPY1x2
+
+
+cgemm_tcopy_L2_M2_END:
+
+       subs    J , J, #1                                               // j--
+       bne     cgemm_tcopy_L2_M2_BEGIN
+
+/*********************************************************************************************/
+
+cgemm_tcopy_L1_BEGIN:
+
+       tst     M, #1
+       ble     cgemm_tcopy_L999
+
+
+cgemm_tcopy_L1_M2_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 + 2 * SIZE *2
+       str     r3, B
+
+       asrs    I, N, #1                                        // I = M / 2
+       ble     cgemm_tcopy_L1_M2_60
+
+
+cgemm_tcopy_L1_M2_40:
+
+       COPY2x1
+       subs I, I, #1
+       bne     cgemm_tcopy_L1_M2_40
+
+cgemm_tcopy_L1_M2_60:
+
+       tst     N , #1
+       ble     cgemm_tcopy_L1_M2_END
+
+       COPY1x1
+
+
+cgemm_tcopy_L1_M2_END:
+
+
+
+cgemm_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
+
diff --git a/kernel/arm/zgemm_tcopy_2_vfpv3.S b/kernel/arm/zgemm_tcopy_2_vfpv3.S
new file mode 100644 (file)
index 0000000..7e27ca6
--- /dev/null
@@ -0,0 +1,245 @@
+/***************************************************************************
+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/07 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 I      r4
+#define        J       r12
+
+#define A_PRE  256
+
+/**************************************************************************************
+* Macro definitions
+**************************************************************************************/
+.macro COPY2x2
+
+       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 COPY1x2
+
+       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 COPY2x1
+
+       fldmiad AO1, { d0 - d3 }
+
+       fstmiad BO1, { d0 - d3 }
+       add     AO1, AO1, #32
+       add     BO1, BO1, M4
+
+.endm
+
+.macro COPY1x1
+
+       fldmiad AO1, { d0 - d1 }
+
+       fstmiad BO2, { d0 - d1 }
+       add     AO1, AO1, #16
+       add     BO2, BO2, #16
+
+.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, #4                                // lda = lda * SIZE * 2
+
+       sub     r4, fp, #128
+       vstm    r4, { d8 - d15}                                 // store floating point registers
+
+       lsl     r4 , M, #4                                      // M * SIZE * 2
+
+       ldr     r3, B
+
+       and     BO2 , N , #-2
+
+       mul     BO2, BO2, r4
+
+       add     BO2 , BO2, r3
+
+       lsl     M4, M, #5                                       // M4 = M * 2 * SIZE * 2
+
+zgemm_tcopy_L2_BEGIN:
+
+       asrs    J, M, #1                                        // J = N / 2
+       ble     zgemm_tcopy_L1_BEGIN
+
+zgemm_tcopy_L2_M2_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 + 4 * SIZE *2
+       str     r3, B
+
+       asrs    I, N, #1                                        // I = M / 2
+       ble     zgemm_tcopy_L2_M2_60
+
+zgemm_tcopy_L2_M2_40:
+
+       COPY2x2
+       subs I, I, #1
+       bne     zgemm_tcopy_L2_M2_40
+
+zgemm_tcopy_L2_M2_60:
+
+       tst     N , #1
+       ble     zgemm_tcopy_L2_M2_END
+
+       COPY1x2
+
+
+zgemm_tcopy_L2_M2_END:
+
+       subs    J , J, #1                                               // j--
+       bne     zgemm_tcopy_L2_M2_BEGIN
+
+/*********************************************************************************************/
+
+zgemm_tcopy_L1_BEGIN:
+
+       tst     M, #1
+       ble     zgemm_tcopy_L999
+
+
+zgemm_tcopy_L1_M2_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 + 2 * SIZE *2
+       str     r3, B
+
+       asrs    I, N, #1                                        // I = M / 2
+       ble     zgemm_tcopy_L1_M2_60
+
+
+zgemm_tcopy_L1_M2_40:
+
+       COPY2x1
+       subs I, I, #1
+       bne     zgemm_tcopy_L1_M2_40
+
+zgemm_tcopy_L1_M2_60:
+
+       tst     N , #1
+       ble     zgemm_tcopy_L1_M2_END
+
+       COPY1x1
+
+
+zgemm_tcopy_L1_M2_END:
+
+
+
+zgemm_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
+